CUDA_VISIBLE_DEVICES=0 python train.py \ | |
--dataset_config config/tvrranking_data_config.json \ | |
--use_interal_vr_scores \ | |
--model_config config/model_config.json \ | |
--eval_tasks_at_training VCMR VR \ | |
--bsz 64 \ | |
--use_extend_pool 500 \ | |
--neg_video_num 0 \ | |
--exp_id reproduce \ | |
--max_vcmr_video 2 \ | |
--similarity_measure general \ | |
--eval_num_per_epoch 1 | |
# qsub -I -l select=1:ngpus=1 -P gs_slab -q slab_gpu8 | |
# cd 11_TVR-Ranking/CONQUER/; conda activate py11; sh run_general.sh | |