I have trained a autoencoder with Early stopping using different epochs but for a certain epoch using early stopping I am getting a minimum loss. How to access that model from that specific address and save it. Below is the picture for reference-
I want to access the model from the specified address and then save it.
Thanks in advance!!!
CodePudding user response:
I am assuming you want to save the model at the time the training early stops.
Refer to:
https://keras.io/api/models/model_saving_apis/
https://keras.io/api/callbacks/model_checkpoint/
A simple model.save()
should work (you can do so after the model.fit()
) or if you want to have checkpoints every time the loss reaches a minimum then use the modelCheckpoint
callback.