首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >Keras在经过额外的持续训练后正确地保存检查点--初始阶段

Keras在经过额外的持续训练后正确地保存检查点--初始阶段
EN

Stack Overflow用户
提问于 2017-07-13 13:13:00
回答 1查看 721关注 0票数 1

当我决定训练新的模型并保存检查点时,ModelCheckpoint工作得很好,因为我希望这些检查点被保存。然而,当我决定为n更多的时代训练同样的模型时,我的问题就出现了。问题是,划时代被重置为0,这将产生一些模型检查点名称如下:

代码语言:javascript
复制
/checkpoints
    checkpoint-01-0.24.h5
    checkpoint-02-0.34.h5
    checkpoint-03-0.37.h5
              .
              .
    checkpoint-m-0.68.h5
    checkpoint-01-0.71.h5
    checkpoint-02-0.73.h5
    checkpoint-03-0.74.h5
              .
              .
    checkpoint-n-0.85.h5

正如你所看到的,时代会被重置。我想要实现的是,在以前的迭代中获取所有历元的数目,并将其添加到新的历次中,得到如下所示:

代码语言:javascript
复制
    checkpoint-(m + 01)-0.71.h5
    checkpoint-(m + 02)-0.73.h5
    checkpoint-(m + 03)-0.74.h5
              .
              .
    checkpoint-(m + n)-0.85.h5
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2017-07-15 08:11:40

由于您可以在医生中读取.fit()函数,有一个参数正是这样做的:

开始训练的initial_epoch:时代(用于恢复以前的训练)

因此,只需添加:

代码语言:javascript
复制
model.fit(..., initial_epoch=m)

就像你的例子一样,m是第一个运行的时代。

我希望这有帮助:)

票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/45081682

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档