From ec14fec89fb90745144411ae8669c601dffc7c10 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 14 Nov 2023 10:06:47 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 8bd6fb47..b51306f6 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -1198,9 +1198,9 @@ def run_average_models(ckpt_dir_s_path, day_night='NIGHT', l1b_andor_l2='BOTH',
for k, w in enumerate(m):
model_lyrs[k].append(w)
for lyr in model_lyrs:
- nda = np.stack(lyr)
+ nda = np.stack(lyr, axis=-1)
print(nda.shape)
- avg = np.mean(nda)
+ avg = np.mean(nda, axis=-1)
print(avg.shape)
avg_model_weights.append(nda)
--
GitLab