From d5cd71c31ba490ff411d0aefa27514bf164ccd65 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 31 Aug 2023 17:09:06 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloud_opd_fcn_abi.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py
index eb58b5bc..199f8ab9 100644
--- a/modules/deeplearning/cloud_opd_fcn_abi.py
+++ b/modules/deeplearning/cloud_opd_fcn_abi.py
@@ -46,6 +46,7 @@ DO_ZERO_OUT = False
CACHE_FILE = ''
USE_EMA = True
EMA_OVERWRITE_FREQUENCY = 5
+EMA_MOMENTUM = 0.99
# setup scaling parameters dictionary
mean_std_dct = {}
@@ -498,6 +499,7 @@ class SRCNN:
optimizer = tf.keras.optimizers.Adam(learning_rate=self.learningRateSchedule,
use_ema=USE_EMA,
+ ema_momentum=EMA_MOMENTUM,
ema_overwrite_frequency=EMA_OVERWRITE_FREQUENCY)
self.optimizer = optimizer
--
GitLab