polish run.sh

2020-04-11 02:53:55 +08:00 · 2020-04-11 02:53:55 +08:00 · 6e23365d87
parent 749c3d4168
commit 6e23365d87
1 changed files with 7 additions and 0 deletions
--- a/tools/run.sh
+++ b/tools/run.sh
@ -0,0 +1,7 @@
+export PYTHONPATH=$(dirname "$PWD"):$PWD:$PYTHONPATH
+
+python -m paddle.distributed.launch \
+    --selected_gpus="0,1,2,3" \
+    --log_dir=log_ResNet50 \
+    tools/train.py \
+        -c ./configs/ResNet/ResNet50.yaml