diff --git a/tools/train.sh b/tools/train.sh index 412f1081e3c4b5d9d4182fc7da10ccf97af4c071..646790287448f82ead9d9f0140d23a941156edfd 100755 --- a/tools/train.sh +++ b/tools/train.sh @@ -1,8 +1,8 @@ #!/usr/bin/env bash # for single card train -# python3.7 tools/train.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml +# python tools/train.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml # for multi-cards train export CUDA_VISIBLE_DEVICES=0,1,2,3 -python3.7 -m paddle.distributed.launch --gpus="0,1,2,3" tools/train.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml +python -m paddle.distributed.launch --gpus="0,1,2,3" tools/train.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml