【问题标题】:How to extract and save the model from Callback history in TensorFlow如何从 TensorFlow 中的回调历史中提取和保存模型
【发布时间】:2021-09-22 05:53:31
【问题描述】:

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

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

提前致谢!!!

【问题讨论】:

  • 你从那个特定地址是什么意思?这里的地址是什么?
  • 什么是 0x7f325...?在这个
  • 您无法在 python 中访问这些地址。不知道这里有什么意义。

标签: tensorflow keras deep-learning tensorflow2.0


【解决方案1】:

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

参考:

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

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

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

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2014-11-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2019-08-30
    • 2018-06-27
    相关资源
    最近更新 更多