|
|
#!/bin/bash |
|
|
|
|
|
DATA_DIR=/disk/gjw/msr-vtt |
|
|
OUT_DIR=./runs/simple_msrvtt |
|
|
|
|
|
python simple_train.py \ |
|
|
--do_train \ |
|
|
--datatype msrvtt \ |
|
|
--output_dir "$OUT_DIR" \ |
|
|
--train_csv "$DATA_DIR/MSRVTT_train.9k.csv" \ |
|
|
--val_csv "$DATA_DIR/MSRVTT_JSFUSION_test.csv" \ |
|
|
--data_path "$DATA_DIR/MSRVTT_data.json" \ |
|
|
--features_path "$DATA_DIR/compressed_videos" \ |
|
|
--epochs 5 \ |
|
|
--batch_size 64 \ |
|
|
--batch_size_val 512 \ |
|
|
--max_words 32 \ |
|
|
--max_frames 12 \ |
|
|
--slice_framepos 2 \ |
|
|
--loose_type \ |
|
|
--num_thread_reader 2 \ |
|
|
--lr 1e-4 \ |
|
|
--coef_lr 1e-3 \ |
|
|
--freeze_layer_num 0 \ |
|
|
--linear_patch 2d \ |
|
|
--sim_header meanP \ |
|
|
--pretrained_clip_name ViT-B/32 \ |
|
|
--expand_msrvtt_sentences \ |
|
|
--n_display 50 |
