tensorboard 恢复预训练模型,在此基础上继续训练(对于.ckpt,.h)!

一、保存.ckpt模型

保存模型,max_to_keep指的是保存最近的10个模型

  self.saver = tf.train.Saver(tf.global_variables(), max_to_keep=10)

放在训练代码的最后,ckpt_file:需要保存ckpt的路径:

   self.saver.save(self.sess, ckpt_file, global_step=epoch)

二,恢复.ckpt模型

如果全部恢复的话代码如下,self.initial_weigh为你的.ckpt文件路径,我这里是:

"./checkpoint/yolov3_coco_demo.ckpt"

恢复代码路径:

self.loader = tf.train.Saver()
 self.loader.restore(self.sess, self.initial_weight)
全部评论

相关推荐

夏目LTH:这个真的很看运气,多投吧。我从去年十二月底一直投到现在,之前没一个offer,结果两周前投的一家面试官聊的特别好,直接速通offer还给的SP待遇开的比我期望都够。
我的求职进度条
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务