tensorflow模型保存
首先调用方法:
saver = tf.train.Saver()
保存模型:
saver.save(sess, "model/model.ckpt")
注意,要写上文件路径,直接写文件名会报错
加载路径:
saver.restore(sess, "model/model.ckpt-24000")
这里要加上训练的轮数,在你的model里会出现,这个问题困扰了很多人,最终解决
首先调用方法:
saver = tf.train.Saver()
保存模型:
saver.save(sess, "model/model.ckpt")
注意,要写上文件路径,直接写文件名会报错
加载路径:
saver.restore(sess, "model/model.ckpt-24000")
这里要加上训练的轮数,在你的model里会出现,这个问题困扰了很多人,最终解决