From ac3f8d37e400e2a04ae63ca471e6b6560af0d736 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 29 Jan 2021 11:08:29 -0600
Subject: [PATCH] snapshot...

---
 modules/deeplearning/cloudheight_mgpus.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/modules/deeplearning/cloudheight_mgpus.py b/modules/deeplearning/cloudheight_mgpus.py
index 98210dd4..1d458ba6 100644
--- a/modules/deeplearning/cloudheight_mgpus.py
+++ b/modules/deeplearning/cloudheight_mgpus.py
@@ -16,13 +16,14 @@ CACHE_DATA_IN_MEM = True
 CACHE_GFS = True
 DISK_CACHE = True
 
-PROC_BATCH_SIZE = 60
 NumLabels = 1
-BATCH_SIZE = 256
 NUM_EPOCHS = 200
-GLOBAL_BATCH_SIZE = BATCH_SIZE * 3
-PROC_BATCH_BUFFER_SIZE = 50000
 
+PROC_BATCH_SIZE = 60
+BATCH_SIZE = 256  # Per replica
+BATCH_SIZE = BATCH_SIZE * 3
+
+PROC_BATCH_BUFFER_SIZE = 50000
 
 TRACK_MOVING_AVERAGE = False
 
-- 
GitLab