snapshot...

modules/deeplearning/unet_l1b_l2.py

@@ -23,7 +23,7 @@ else:
     NumLogits = NumClasses
 
 BATCH_SIZE = 128
-NUM_EPOCHS = 40
+NUM_EPOCHS = 60
 
 TRACK_MOVING_AVERAGE = False
 EARLY_STOP = True
@@ -417,7 +417,7 @@ class UNET:
         momentum = 0.99
 
         # num_filters = len(self.train_params) * 4
-        num_filters = self.n_chans * 4
+        num_filters = self.n_chans * 12
 
         input_2d = self.inputs[0]
         conv = tf.keras.layers.Conv2D(num_filters, kernel_size=5, strides=1, padding=padding, activation=None)(input_2d)