From 87200f5cd2a439e28db61a6892531c723db9ba8b Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 24 Oct 2022 16:54:05 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index db2fdc69..382654bb 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -719,7 +719,7 @@ class IcingIntensityFCN:
# optimizer = tfa.optimizers.MovingAverage(optimizer)
self.ema = tf.train.ExponentialMovingAverage(decay=0.9999)
self.ema.apply(self.model.trainable_variables)
- self.ema_trainable_variables = self.ema.average(self.model.trainable_variables)
+ self.ema_trainable_variables = [self.ema.average(var) for var in self.model.trainable_variables]
self.optimizer = optimizer
self.initial_learning_rate = initial_learning_rate
--
GitLab