diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index b186f13ee814f7688e4dc59accf818a0a2426ff6..401a1461c5081eb53112709ec157c0ca8c436a43 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1190,7 +1190,11 @@ 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 ******************************************') z_model_weights = zip(model_weight_s) - print('test: ', len(z_model_weights)) + print('test') + avg_model_weights = [] + for w in z_model_weights: + avg_model_weights(np.mean(w)) + print('***', len(avg_model_weights)) avg_model_weights = [] for m in model_weight_s: