Home > Mobile >  How to extract and save the model from Callback history in TensorFlow
How to extract and save the model from Callback history in TensorFlow

Time:09-23

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- enter image description here

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.

  • Related