run_dali.sh 304 B

12345678910
  1. #!/usr/bin/env bash
  2. export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
  3. export FLAGS_fraction_of_gpu_memory_to_use=0.80
  4. python3.7 -m paddle.distributed.launch \
  5. --gpus="0,1,2,3,4,5,6,7" \
  6. ppcls/static/train.py \
  7. -c ./ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml \
  8. -o Global.use_dali=True