diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py index 1513f35a381de7318f2be15a58196ead95c38f07..6235b947d424b9d72b250e5bb7ae8cac52faf14a 100644 --- a/modules/deeplearning/unet_l1b_l2.py +++ b/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)