File size: 900 Bytes
6dd9459 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
python baselines/crossmodal_moment_localization/train.py \
--train_path data/TVR_Ranking/train_top40.json \
--val_path data/TVR_Ranking/val.json \
--test_path data/TVR_Ranking/test.json \
--corpus_path data/TVR_Ranking/video_corpus.json \
--desc_bert_path data/features/query_bert.h5 \
--vid_feat_path data/features/tvr_i3d_rgb600_avg_cl-1.5.h5 \
--sub_bert_path data/features/tvr_sub_pretrained_w_sub_query_max_cl-1.5.h5\
--dset_name=tvr \
--eval_split_name=val \
--nms_thd=-1 \
--results_root=results \
--clip_length=1.5 \
--vid_feat_size=1024 \
--ctx_mode=video_sub_tef \
--max_ctx_l=128 \
--max_pred_l=16 \
--eval_num_per_epoch=2 \
--n_epoch=100 \
--exp_id=top40 \
--model_name=XML
# qsub -I -l select=1:ngpus=1 -P gs_slab -q gpu8
# cd 11_TVR-Ranking/TVRetrieval/; conda activate py11; sh run_top40.sh |