From d521effbd47980bee45ec95851e14020eee70f4d Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 20 May 2022 12:08:08 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/unet_l1b_l2.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py
index 1513f35a..6235b947 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)
--
GitLab