一、保存.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)