/Project_Gen

Primary LanguageJupyter Notebook

Project_Gen

python train_SGPT_V2_Sentence.py
--lr=0.00001
--checkpoint_dir='checkpoints/efvlegpt2rs18/m18_v1_z_qf_'
--dataset_type='m18'
--tokenizer_ver='gpt2v1'
--model_ver='efvlegpt2rs18'
--model_subver='v1'
--vis_pos_emb='zeroes'
--batch_size=40

Other commands:
python train_SGPT_V2_Sentence.py --lr=0.00001 --checkpoint_dir='checkpoints/efvlegpt2rs18/m18_v1_z_qf_' --dataset_type='m18' --tokenizer_ver='gpt2v1' --model_ver='efvlegpt2rs18' --model_subver='v1' --vis_pos_emb='zeroes' --batch_size=40

Dataset

EndoVis18-VQA