From 571dc43288cc9c58e110651cdd029323d389ce8a Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 13 Jul 2021 12:24:46 -0500
Subject: [PATCH] minor...

---
 modules/deeplearning/icing_cnn.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index a65f86d1..0c7add6d 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -16,8 +16,13 @@ CACHE_DATA_IN_MEM = False
 
 PROC_BATCH_SIZE = 4096
 PROC_BATCH_BUFFER_SIZE = 50000
+
 NumClasses = 2
-NumLogits = 1
+if NumClasses == 2:
+    NumLogits = 1
+else:
+    NumLogits = NumClasses
+
 BATCH_SIZE = 128
 NUM_EPOCHS = 100
 
-- 
GitLab