From d6f3148d6f22c83324fdc47fa12784f147effd8d Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 29 Jan 2021 10:40:46 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloudheight_mgpus.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/cloudheight_mgpus.py b/modules/deeplearning/cloudheight_mgpus.py
index 65c8879f..98210dd4 100644
--- a/modules/deeplearning/cloudheight_mgpus.py
+++ b/modules/deeplearning/cloudheight_mgpus.py
@@ -669,7 +669,7 @@ class CloudHeightNN:
with tf.GradientTape() as tape:
pred = self.model(inputs, training=True)
loss = self.loss(labels, pred)
- loss = tf.nn.compute_average_loss(loss, global_batch_size=GLOBAL_BATCH_SIZE)
+ loss = tf.nn.compute_average_loss(loss, global_batch_size=BATCH_SIZE)
total_loss = loss
if len(self.model.losses) > 0:
reg_loss = tf.math.add_n(self.model.losses)
--
GitLab