diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 401a1461c5081eb53112709ec157c0ca8c436a43..be066c64852c308ae13cc3afbae472c5d788b818 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1192,8 +1192,10 @@ def run_average_models(ckpt_dir_s_path, day_night='NIGHT', l1b_andor_l2='BOTH', z_model_weights = zip(model_weight_s) print('test') avg_model_weights = [] - for w in z_model_weights: - avg_model_weights(np.mean(w)) + for lyr in z_model_weights: + print(len(lyr)) + for w in lyr: + avg_model_weights(np.mean(w)) print('***', len(avg_model_weights)) avg_model_weights = []