2021-06-11 12:17:56 +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
|
2022-01-13 18:24:41 +08:00
|
|
|
export CUDA_VISIBLE_DEVICES=0,1,2,3
|
2021-09-26 15:05:13 +08:00
|
|
|
python3.7 -m paddle.distributed.launch --gpus="0,1,2,3" tools/train.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml
|
