From cb090084c78df8dd3a5967969886ae7ce0e4455d Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 21 Oct 2021 15:52:33 -0500
Subject: [PATCH] employ ExponentialWeigthedAverage to training
---
modules/deeplearning/icing_cnn.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index e13d3cf4..51f40292 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -682,6 +682,7 @@ class IcingIntensityNN:
self.optimizer.apply_gradients(zip(gradients, self.model.trainable_variables))
if TRACK_MOVING_AVERAGE:
self.ema.apply(self.model.trainable_variables)
+ # TODO: This doesn't seem to work
# for var in self.model.trainable_variables:
# var.assign(self.ema.average(var))
--
GitLab