PaddleClas/ppcls/static/run_dali.sh

12 lines
305 B
Bash
Raw Normal View History

#!/usr/bin/env bash
2021-07-15 11:19:38 +08:00
export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
export FLAGS_fraction_of_gpu_memory_to_use=0.80
python3.7 -m paddle.distributed.launch \
2021-07-15 11:19:38 +08:00
--gpus="0,1,2,3,4,5,6,7" \
2021-10-12 07:38:02 +00:00
ppcls/static/train.py \
-c ./ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml \
-o Global.use_dali=True