From 6f8d142edcb0fe9263f6a2fe53383a6bd7e8256d Mon Sep 17 00:00:00 2001 From: Christopher Teubert Date: Thu, 13 Jul 2023 11:04:37 -0700 Subject: [PATCH] Fix issue with savepts --- src/progpy/data_models/lstm_model.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/src/progpy/data_models/lstm_model.py b/src/progpy/data_models/lstm_model.py index 58ec924e..853338f7 100644 --- a/src/progpy/data_models/lstm_model.py +++ b/src/progpy/data_models/lstm_model.py @@ -537,9 +537,7 @@ def from_data(cls, inputs, outputs, event_states=None, t_met=None, **kwargs): from tensorflow import keras # Build model - callbacks = [ - keras.callbacks.ModelCheckpoint("best_model.keras", save_best_only=True) - ] + callbacks = [ ] if params['early_stop']: callbacks.append(keras.callbacks.EarlyStopping(**params['early_stop.cfg'])) @@ -592,8 +590,6 @@ def from_data(cls, inputs, outputs, event_states=None, t_met=None, **kwargs): workers=params['workers'], use_multiprocessing=(params['workers'] > 1)) - model = keras.models.load_model("best_model.keras") - # Split model into separate models n_state_layers = params['layers'] + 1 + (params['dropout'] > 0) + (params['normalize']) output_layer_input = keras.layers.Input(model.layers[n_state_layers-1].output.shape[1:])