CUDA_VISIBLE_DEVICES=0 python train.py \ | |
--dataset_config config/tvrranking_data_config.json \ | |
--model_config config/model_config.json \ | |
--eval_tasks_at_training VCMR \ | |
--use_interal_vr_scores \ | |
--bsz 64 \ | |
--use_extend_pool 500 \ | |
--neg_video_num 0 \ | |
--exp_id debug \ | |
--bsz 5 \ | |
--max_vcmr_video 10 \ | |
--num_workers 4 | |
# qsub -I -l select=1:ngpus=1 -P gs_slab -q gpu8 | |
# cd 11_TVR-Ranking/CONQUER/; conda activate py11; sh run.sh |