【问题标题】:checkpointing the model every iteration每次迭代都检查模型
【发布时间】:2016-06-20 23:14:02
【问题描述】:

在 TensorFlow 中每第 N 次迭代检查点(即保存快照)模型的最佳做法是什么?

显然,可以使用自定义逻辑调用tf.train.Saver.save,但想知道是否有内部机制?

【问题讨论】:

    标签: python tensorflow


    【解决方案1】:

    我会根据我希望保存检查点的频率来做这样的事情:

    if (epoch % N) == (N - 1):
        saveto=CkptDir+"ckpt/model"+str(N)+".ckpt"
        save_path = saverRun.save(session, saveto)
    

    这将保存一个 N epoch 的检查点文件。

    【讨论】:

      【解决方案2】:

      找到解决方案here。可以设置max_to_keepkeep_checkpoint_every_n_hours 以获得所需的节省。

      【讨论】:

        猜你喜欢
        • 2016-08-19
        • 1970-01-01
        • 2011-10-11
        • 1970-01-01
        • 2020-08-27
        • 1970-01-01
        • 1970-01-01
        • 2020-04-18
        • 1970-01-01
        相关资源
        最近更新 更多