diff --git a/modules/deeplearning/unet.py b/modules/deeplearning/unet.py
index 16e44f6f630fc896a5aab2812dc7237981141034..bab6c4d57ca1a18561f7ee57b83aebb59f4d93c7 100644
--- a/modules/deeplearning/unet.py
+++ b/modules/deeplearning/unet.py
@@ -15,7 +15,7 @@ import h5py
 
 LOG_DEVICE_PLACEMENT = False
 
-PROC_BATCH_SIZE = 1024
+PROC_BATCH_SIZE = 50
 PROC_BATCH_BUFFER_SIZE = 50000
 
 NumClasses = 2