mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-24 06:20:55 +00:00
6 lines
167 B
Bash
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
|