diff --git a/ppcls/static/run_dali.sh b/ppcls/static/run_dali.sh index 559c4b345adcdc2fba0bab77247822d2d104ebc1..8b33b28d28d0b83a163244495a6076fb63fd4a02 100644 --- a/ppcls/static/run_dali.sh +++ b/ppcls/static/run_dali.sh @@ -1,10 +1,10 @@ #!/usr/bin/env bash -export CUDA_VISIBLE_DEVICES="0,1,2,3" +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 \ - --gpus="0,1,2,3" \ + --gpus="0,1,2,3,4,5,6,7" \ ppcls/static//train.py \ -c ./ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml \ -o Global.use_dali=True