From 88daf8cf034872b6dabbf62136f2fefccbbab4d4 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 24 Oct 2022 18:03:38 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 382654bb..ca9792d7 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -992,7 +992,11 @@ class IcingIntensityFCN:
def restore(self, ckpt_dir):
- ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)
+ if TRACK_MOVING_AVERAGE:
+ ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model, averaged_weights=self.model.trainable_variables)
+ else:
+ ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)
+
ckpt_manager = tf.train.CheckpointManager(ckpt, ckpt_dir, max_to_keep=3)
ckpt.restore(ckpt_manager.latest_checkpoint)
--
GitLab