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