From 4fd818b0ea203601ac8840eba17eea9c8ff94be8 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 14 Nov 2023 09:57:00 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 13 ++++++++-----
1 file changed, 8 insertions(+), 5 deletions(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 1d561907..0feb0719 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -1190,13 +1190,16 @@ def run_average_models(ckpt_dir_s_path, day_night='NIGHT', l1b_andor_l2='BOTH',
model_weight_s.append(k_model.get_weights())
print('done loading models ******************************************')
+ num_model_lyrs = len(model_weight_s[0])
+ model_lyrs = [[] for k in range(num_model_lyrs)]
+
avg_model_weights = []
for m in model_weight_s:
- print(len(m))
- for w in m:
- print(w.shape)
- # avg_model_weights.append(np.mean(w))
- print('--------------------------------------')
+ for k, w in enumerate(m):
+ model_lyrs[k].append(w)
+ for lyr in model_lyrs:
+ nda = np.stack(lyr)
+ avg_model_weights.append(nda)
# -- Make a new model for the averaged weights
new_model = IcingIntensityFCN(day_night=day_night, l1b_or_l2=l1b_andor_l2, use_flight_altitude=use_flight_altitude)
--
GitLab