update

Author: YunYang1994

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