From 78733d7da7ab809f6a768971fd0ee29317798128 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 27 Oct 2022 14:46:55 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/icing_fcn.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 1929d96e..0905a2e8 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -1001,8 +1001,9 @@ class IcingIntensityFCN:
ckpt.restore(ckpt_manager.latest_checkpoint)
- for idx, var in enumerate(self.model.trainable_variables):
- var.assign(self.ema_trainable_variables[idx])
+ if TRACK_MOVING_AVERAGE:
+ for idx, var in enumerate(self.model.trainable_variables):
+ var.assign(self.ema_trainable_variables[idx])
self.reset_test_metrics()
--
GitLab