File size: 127 Bytes
04fbff5
 
 
 
 
 
1
2
3
4
5
6
NUM_GPU=$1
CFG=$2
PORT=$3
python -m torch.distributed.launch \
--nproc_per_node $NUM_GPU \
--master_port $PORT train.py -c $CFG