Add File
This commit is contained in:
7
tools/scripts/dist_test.sh
Normal file
7
tools/scripts/dist_test.sh
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
|
||||||
|
set -x
|
||||||
|
NGPUS=$1
|
||||||
|
PY_ARGS=${@:2}
|
||||||
|
|
||||||
|
python -m torch.distributed.launch --nproc_per_node=${NGPUS} test.py --launcher pytorch ${PY_ARGS}
|
||||||
Reference in New Issue
Block a user