PaddleClas/train.sh

8 lines
233 B
Bash
Raw Normal View History

2021-09-26 15:05:13 +08:00
#!/usr/bin/env bash
# for single card train
# python3.7 tools/train.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml
# for multi-cards train
python3.7 -m paddle.distributed.launch --gpus="0" tools/train.py -c ./MobileNetV2.yaml