diff --git a/scripts/slurm_dist_train.sh b/scripts/slurm_dist_train.sh deleted file mode 100644 index 1d3d505f3..000000000 --- a/scripts/slurm_dist_train.sh +++ /dev/null @@ -1,11 +0,0 @@ -#!/usr/bin/env sh - - -main_file=$1 -config_file=$2 - -python $main_file --local_rank $SLURM_PROCID --world_size $SLURM_NPROCS --host $HOST --port 29500 --config $config_file - -# how to run this script -# exmaple: -# HOST=IP_ADDR srun ./scripts/slurm_dist_train.sh ./examples/train_vit_2d.py ./configs/vit/vit_2d.py \ No newline at end of file