eP-ALM / run_scripts /float32 /image /ePALM_pt_L_vqa.sh
mshukor
init
3eb682b
raw
history blame
671 Bytes
# visopt_vqa_vlt5_stok_6clst_l19_31_reptok_nocache_ptl10_ptlr1e3fixed_nomlp_sharedcon_opt_6_7b_vitl_stdsplit_ep64
config=./configs/image/ePALM_pt_L_vqa.yaml
data_dir=data/vl_adapter/vlt5_dataset
output_dir=logs/epalm/ePALM_pt_L_vqa
torchrun --nproc_per_node=8 --master_addr="localhost" --master_port=12302 float32/vqa.py \
--config $config \
--output_dir $output_dir \
--data_dir $data_dir \
--save_best \
--text_model facebook/opt-6.7b --vision_model vit_large_patch16_224 \
# --resume --checkpoint $WORK/logs/epalm/ePALM_pt_L_vqa/checkpoint_last.pth
# --checkpoint /data/mshukor/logs/epalm/ePALM_pt_L_vqa/checkpoint_last.pth --evaluate --open_ended_eval \