如何从 TensorFlow 中的回调历史中提取和保存模型

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

我已经使用不同的时期训练了一个提前停止的自动编码器,但是对于使用提前停止的某个时期,我得到的损失最小。如何从该特定地址访问该模型并保存它。下图供参考——

我想从指定地址访问模型然后保存。

提前致谢!!!

我假设您想在训练提前停止时保存模型。

参考:

https://keras.io/api/models/model_saving_apis/

https://keras.io/api/callbacks/model_checkpoint/

一个简单的 model.save() 应该可以工作(你可以在 model.fit() 之后这样做)或者如果你想在每次损失达到最小值时都有检查点然后使用 modelCheckpoint 回调.