diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 0feb07194f96086f20911a392235621d9fb4c3c4..8bd6fb4754fbad7356bf7dfa535ef493262af0c6 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1199,6 +1199,9 @@ def run_average_models(ckpt_dir_s_path, day_night='NIGHT', l1b_andor_l2='BOTH', model_lyrs[k].append(w) for lyr in model_lyrs: nda = np.stack(lyr) + print(nda.shape) + avg = np.mean(nda) + print(avg.shape) avg_model_weights.append(nda) # -- Make a new model for the averaged weights