snapshot...

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)