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)
全部评论

相关推荐

线性袋鼠:别听牛客上一帮伪人在那说,小厂不能去,必须去大厂,听他们放屁吧。学院本+一些一本最终的归宿就是中小厂,大厂那么好进吗
我的实习日记
点赞 评论 收藏
分享
今天 14:46
门头沟学院 Java
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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