8 lines
271 B
Bash
Executable File
8 lines
271 B
Bash
Executable File
#!/usr/bin/env bash
|
|
CONFIG=$1
|
|
GPUS=$2
|
|
PORT=${PORT:-29500}
|
|
# cat /proc/193481/cmdline
|
|
PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
|
|
python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=63667 \
|
|
$(dirname "$0")/train.py $CONFIG --launcher pytorch ${@:3} |