Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Train command gives error #10

Open
zjjj opened this issue Sep 1, 2020 · 1 comment
Open

Train command gives error #10

zjjj opened this issue Sep 1, 2020 · 1 comment

Comments

@zjjj
Copy link

zjjj commented Sep 1, 2020

The command in "Training a New Model" section (see below) will give an error message No such file or directory: 'config/pytorch_model.bin'. Should we add --load_model trained_model/pytorch_model to the command, as I noticed that there's a pytorch_model.bin file in the trained_model folder downloaded from google drive?

python code/KBQA_Runner.py  \
        --train_folder  data/train_CQ \
        --dev_folder data/dev_CQ \
        --test_folder data/test_CQ \
        --KB_file data/CQ/kb_cache.json \
        --M2N_file data/CQ/m2n_cache.json \
        --QUERY_file data/CQ/query_cache.json \
        --vocab_file data/CQ/vocab.txt \
        --output_dir trained_model/CQ \
        --config config/bert_config_CQ.json \
        --gpu_id 1\
        --save_model my_model \
        --max_hop_num 2 \
        --num_train_epochs 100 \
        --do_train 1\
        --do_eval 1\
        --do_policy_gradient 1\
        --learning_rate 1e-5 \
        --train_limit_number 150 \
@lanyunshi
Copy link
Owner

Hi,

Sorry for the late reply.
The file you are looking for is the pre-trained model of BERT. It exists under the trained-model folder.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants