| python train.py \ |
| --results_path results/tvr_ranking \ |
| --train_path data/TVR_Ranking/train_top01.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 \ |
| --video_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 \ |
| --n_epoch 4000 \ |
| --eval_num_per_epoch 0.05 \ |
| --seed 2024 \ |
| --bsz 512 \ |
| --exp_id top01 |
|
|
| |
| |