File size: 525 Bytes
a638e43 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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
|