update

YunYang1994 · YunYang1994 · commit 44a938039184 · 2020-02-25T16:23:48.000+08:00
diff --git a/7-Utils/multi_gpu_train.py b/7-Utils/multi_gpu_train.py
@@ -18,7 +18,7 @@
 from tensorflow.keras.preprocessing.image import ImageDataGenerator
 from tensorflow.keras import applications
 
-os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"
+os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3"
 
 EPOCHS       = 40
 SCORE_THRESH = 0.8
