diff --git a/modules/deeplearning/unet.py b/modules/deeplearning/unet.py
index 4b6f930250e6286fb782a7525cd9de624ac5610a..3d7961deac188df86d2b705f4183ff07164894b6 100644
--- a/modules/deeplearning/unet.py
+++ b/modules/deeplearning/unet.py
@@ -25,14 +25,14 @@ BATCH_SIZE = 128
 NUM_EPOCHS = 100
 
 TRACK_MOVING_AVERAGE = False
-EARLY_STOP = False
+EARLY_STOP = True
 
 TRIPLET = False
 CONV3D = False
 
-NOISE_TRAINING = False
+NOISE_TRAINING = True
 NOISE_STDDEV = 0.10
-DO_AUGMENT = False
+DO_AUGMENT = True
 
 img_width = 16