diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index be066c64852c308ae13cc3afbae472c5d788b818..6a457584a070bd40aedf8ca6d6ee6c908bcc34d9 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1189,14 +1189,6 @@ def run_average_models(ckpt_dir_s_path, day_night='NIGHT', l1b_andor_l2='BOTH', k_model = model.model model_weight_s.append(k_model.get_weights()) print('done loading models ******************************************') - z_model_weights = zip(model_weight_s) - print('test') - avg_model_weights = [] - 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 = [] for m in model_weight_s: @@ -1204,7 +1196,7 @@ def run_average_models(ckpt_dir_s_path, day_night='NIGHT', l1b_andor_l2='BOTH', for w in m: print(w.shape) avg_model_weights.append(np.mean(w)) - + print('--------------------------------------') # -- Make a new model for the averaged weights new_model = IcingIntensityFCN(day_night=day_night, l1b_or_l2=l1b_andor_l2, use_flight_altitude=use_flight_altitude)