You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
9 lines
302 B
9 lines
302 B
#!/usr/bin/env bash
|
|
|
|
# for single card train
|
|
# python3.7 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
|