simple_train.sh

#!/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