snapshot...

modules/deeplearning/cloudheight.py

@@ -16,9 +16,9 @@ CACHE_DATA_IN_MEM = True
 CACHE_GFS = True
 DISK_CACHE = True
 
-PROC_BATCH_SIZE = 40
+PROC_BATCH_SIZE = 60
 NumLabels = 1
-BATCH_SIZE = 512
+BATCH_SIZE = 1024
 NUM_EPOCHS = 200
 GLOBAL_BATCH_SIZE = BATCH_SIZE * 3
 
@@ -687,10 +687,10 @@ class CloudHeightNN:
         labels = mini_batch[2]
         pred = self.model(inputs, training=False)
         t_loss = self.loss(labels, pred)
-        t_loss = tf.nn.compute_average_loss(t_loss, global_batch_size=GLOBAL_BATCH_SIZE)
 
         self.test_loss(t_loss)
         self.test_accuracy(labels, pred)
+
         return t_loss
 
     @tf.function