diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index fdc99e1b4fd8544201b091341bbca8a1e3affa44..c44153cd496357adf9e5276695c6cfbe8e0d647d 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1240,11 +1240,11 @@ def run_evaluate_static_2(model, data_dct, num_tiles, prob_thresh=0.5, flight_le return preds_dct, probs_dct -def load_model(model_path, day_night='NIGHT', l1b_andor_l2='BOTH', satellite='GOES16', use_flight_altitude=False): +def load_model(model_path, day_night='NIGHT', l1b_andor_l2='BOTH', use_flight_altitude=False): ckpt_dir_s = os.listdir(model_path) ckpt_dir = model_path + ckpt_dir_s[0] - model = IcingIntensityFCN(day_night=day_night, l1b_or_l2=l1b_andor_l2, satellite=satellite, use_flight_altitude=use_flight_altitude) + model = IcingIntensityFCN(day_night=day_night, l1b_or_l2=l1b_andor_l2, use_flight_altitude=use_flight_altitude) model.build_model() model.build_training() model.build_evaluation()