run_SlowFast_MViTv2_S_16x4_PMV_release.sh

#!/bin/bash

cd MViT/
PYTHONPATH=slowfast:$PYTHONPATH \
output_dir=${UPLOAD_SOURCE}

work_path=configs/Kinetics/
ckpt_path=$output_dir

mkdir -p $ckpt_path

CUR_DIR=$(cd $(dirname $0); pwd)
cd $CUR_DIR

python3 tools/run_net.py \
  --cfg $work_path/MVITv2_S_16x4.yaml \
  --init_method tcp://$WORKER_0_HOST:28999 \
  --num_shards ${WORKER_NUM} \
  --shard_id ${WORKER_ID} \
  --opts \
  SOLVER.BASE_LR_SCALE_NUM_SHARDS True \
  DATA.PATH_TO_DATA_DIR ./data_list/PMV \
  DATA.PATH_LABEL_SEPARATOR "," \
  MODEL.NUM_CLASSES 400 \
  DATA_LOADER.NUM_WORKERS 16 \
  NUM_GPUS 8 \
  OUTPUT_DIR $ckpt_path $@