diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index 8a5cc823cb07e17789f36272cd2edab2036c71f4..dd0e66431ec25a89d675d175e53a9fa2594dadb9 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/modules/deeplearning/cloudheight.py
@@ -15,10 +15,10 @@ LOG_DEVICE_PLACEMENT = False
 CACHE_DATA_IN_MEM = True
 CACHE_GFS = True
 
-PROC_BATCH_SIZE = 40
+PROC_BATCH_SIZE = 60
 PROC_BATCH_BUFFER_SIZE = 50000
 NumLabels = 1
-BATCH_SIZE = 384
+BATCH_SIZE = 256
 NUM_EPOCHS = 200