diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index 1728c0762f7f9e887ad779d1973f19b4af514f37..fc0c3f112673f89f89b20d4991efc0a2ca9db6dc 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/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