From adc02cea6fa9c17b499da3b359e96b0115a505e1 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 27 Oct 2021 15:08:34 -0500
Subject: [PATCH] back out some EWA stuff...
---
modules/deeplearning/icing_cnn.py | 2 --
1 file changed, 2 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index afb0cf37..36acfb27 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -690,8 +690,6 @@ class IcingIntensityNN:
self.optimizer.apply_gradients(zip(gradients, self.model.trainable_variables))
if TRACK_MOVING_AVERAGE:
self.ema.apply(self.model.trainable_variables)
- for var in self.model.trainable_variables:
- var.assign(self.ema.average(var))
self.train_loss(loss)
self.train_accuracy(labels, pred)
--
GitLab