Log In Sign Up
Models:
Mathew Swanson
Mathew-S/
Kaggle-UWMGIT
0
Downloads: 1
[f1e01c]: / tools / dist_train.sh

Download this file

10 lines (7 with data), 247 Bytes 

1
2
3
4
5
6
7
8
9
#!/usr/bin/env bash



CONFIG=$1

GPUS=$2

PORT=${PORT:-29500}



PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \

python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \

    $(dirname "$0")/train.py $CONFIG --launcher pytorch ${@:3}