Merge pull request #65 from nasa/bug/lstm_save

Fix issue with saving model

Author: teubert

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:])