snapshot...

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)