mirror of
https://github.com/botastic/SoftGroup.git
synced 2025-10-16 11:45:42 +00:00
7 lines
162 B
Bash
Executable File
7 lines
162 B
Bash
Executable File
#!/usr/bin/env bash
|
|
CONFIG=$1
|
|
GPUS=$2
|
|
PORT=${PORT:-29500}
|
|
|
|
OMP_NUM_THREADS=1 torchrun --nproc_per_node=$GPUS --master_port=$PORT ./train.py --dist $CONFIG ${@:3}
|