为了账号安全,请及时绑定邮箱和手机立即绑定

tensorflow怎么调用ckpt继续训练 ?

tensorflow怎么调用ckpt继续训练 ?

三国纷争 2018-10-14 17:13:44
tensorflow怎么调用ckpt继续训练 
查看完整描述

1 回答

?
智慧大石

TA贡献1946条经验 获得超3个赞

训练完一个模型后,为了以后重复使用,通常我们需要对模型的结果进行保存。如果用Tensorflow去实现神经网络,所要保存的就是神经网络中的各项权重值。建议可以使用Saver类保存和加载模型的结果。

1、使用tf.train.Saver.save()方法保存模型

  • sess: 用于保存变量操作的会话。

  • save_path: String类型,用于指定训练结果的保存路径。

  • global_step: 如果提供的话,这个数字会添加到save_path后面,用于构建checkpoint文件。这个参数有助于我们区分不同训练阶段的结果。

  • 2、使用tf.train.Saver.restore方法价值模型

    • sess: 用于加载变量操作的会话。

    • save_path: 同保存模型是用到的的save_path参数。

    • 下面通过一个代码演示这两个函数的使用方法

    假设保存变量的时候是

checkpoint_filepath='models/train.ckpt'saver.save(session,checkpoint_filepath)则从文件读变量取值继续训练是saver.restore(session,checkpoint_filepath)



查看完整回答
反对 回复 2018-11-11
  • 1 回答
  • 0 关注
  • 2433 浏览

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信