From f7ab1c2dcaf02c57fe53e19e3eda66e92e551306 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <dazhiningsibuqu@163.com>
Date: Thu, 15 Jul 2021 11:19:38 +0800
Subject: [PATCH] Update run_dali.sh

---
 ppcls/static/run_dali.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ppcls/static/run_dali.sh b/ppcls/static/run_dali.sh
index 559c4b34..8b33b28d 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
-- 
GitLab