From cbf3c432ff6810d7d1ade4e60ec9d10dce832091 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 13 Nov 2023 16:19:49 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 15 ++++++++-------
1 file changed, 8 insertions(+), 7 deletions(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index d5607669..b89faa92 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -1188,6 +1188,7 @@ def run_average_models(ckpt_dir_s_path, day_night='NIGHT', l1b_andor_l2='BOTH',
model = load_model(ckpt_dir, day_night=day_night, l1b_andor_l2=l1b_andor_l2, use_flight_altitude=use_flight_altitude)
k_model = model.model
model_weight_s.append(k_model.get_weights())
+ print('done loading models ******************************************')
avg_model_weights = []
for m in model_weight_s:
@@ -1204,13 +1205,13 @@ def run_average_models(ckpt_dir_s_path, day_night='NIGHT', l1b_andor_l2='BOTH',
new_model.build_evaluation()
# -- Save the averaged weights to a new the model
- if not os.path.exists(modeldir):
- os.mkdir(modeldir)
- ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=new_model.model)
- ckpt_manager = tf.train.CheckpointManager(ckpt, modeldir, max_to_keep=3)
-
- new_model.model.set_weights(avg_weights)
- ckpt_manager.save()
+ # if not os.path.exists(modeldir):
+ # os.mkdir(modeldir)
+ # ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=new_model.model)
+ # ckpt_manager = tf.train.CheckpointManager(ckpt, modeldir, max_to_keep=3)
+ #
+ # new_model.model.set_weights(avg_weights)
+ # ckpt_manager.save()
return
--
GitLab