File size: 603 Bytes
3eb682b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# visopt_video_msvd_vqa_reptok_6clst_l19_31_nocache_noeostk_ptl10_ptlr1e5fixed_tformer_8f_bs8_longt_opt_2_7b
config=./configs/video/ePALM_video_qa_msvd.yaml
data_dir=data/MSVD
output_dir=logs/epalm/ePALM_video_qa_msvd
torchrun --nproc_per_node=4 --master_addr="localhost" --master_port=12325 float32/video_vqa.py \
--config $config \
--output_dir $output_dir \
--data_dir $data_dir \
--save_best \
--text_model facebook/opt-2.7b \
# --resume --checkpoint $WORK/logs/epalm/ePALM_video_qa_msvd/checkpoint_last.pth
# --checkpoint $WORK/logs/epalm/ePALM_video_qa_msvd/checkpoint_best.pth --evaluate
|