ColossalAI/tests/test_trainer/test.sh

6 lines
167 B
Bash

#!/usr/bin/env sh
test_file=$1
config_file=$2
python $test_file --local_rank $SLURM_PROCID --world_size $SLURM_NPROCS --host $HOST --port 29500 --config $config_file