From a9d340e44c4562edd36e6134ac84ebeca4cbced8 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 31 Aug 2023 10:34:57 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloud_opd_fcn_abi.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py
index 5b14e79b..40e17a5b 100644
--- a/modules/deeplearning/cloud_opd_fcn_abi.py
+++ b/modules/deeplearning/cloud_opd_fcn_abi.py
@@ -42,6 +42,7 @@ SIGMA = 1.0
DO_ZERO_OUT = False
CACHE_FILE = '/scratch/long/rink/cld_opd_abi_128x128_cache'
USE_EMA = True
+EMA_OVERWRITE_FREQUENCY = 5
# setup scaling parameters dictionary
mean_std_dct = {}
@@ -482,7 +483,9 @@ class SRCNN:
self.learningRateSchedule = tf.keras.optimizers.schedules.ExponentialDecay(initial_learning_rate, decay_steps, decay_rate)
- optimizer = tf.keras.optimizers.Adam(learning_rate=self.learningRateSchedule, use_ema=USE_EMA)
+ optimizer = tf.keras.optimizers.Adam(learning_rate=self.learningRateSchedule,
+ use_ema=USE_EMA,
+ ema_overwrite_frequency=EMA_OVERWRITE_FREQUENCY)
self.optimizer = optimizer
self.initial_learning_rate = initial_learning_rate
--
GitLab