Andrew82106 issues

Repositories
Issues
Comments

Results 1 issues of


                                            Andrew82106

pytorch size mismatch issue

We use the following command for fine-tuning the OKVQA model: ``` python3 -u train_tasks.py --model_version 3 --bert_model=bert-base-uncased \ --from_pretrained=/workspace/ConceptBERT-master/conceptBert/vilbert/data2/kilbert_base_model/pytorch_model_11.bin \ --from_pretrained_conceptBert=/workspace/ConceptBERT-master/conceptBert/outputs/train1_vqa_trained_model/VQA_bert_base_6layer_6conect/pytorch_model_18.bin \ --output_dir=/workspace/ConceptBERT-master/conceptBert/outputs/train2_okvqa_trained_model/ \ --summary_writer =/workspace/ConceptBERT-master/conceptBert/outputs/tensorboards/ \ --num_workers 0 \...