VBench/vbench/third_party/amt/scripts/train.sh

NUM_GPU=$1
CFG=$2
PORT=$3
python -m torch.distributed.launch \
--nproc_per_node $NUM_GPU \
--master_port $PORT train.py -c $CFG