So I used the Model.save function from tensorflow.keras.models to save 2 models. However, after I load these models, Model.fit stops training the w