tensorflow保存训练好的深度学习模型

最近在看MKR的算法,复现跑通了以后,把训练好的模型保存下来

加入以下代码即可

saver = tf.train.Saver()
saver.save(sess, '../model/MKRmodel.ckpt')
print('模型已保存')

上面的代码要写到和sess.run()一同缩进的地方,不然会报错。

中间代码太长了,第一次写到sess.run()前面那个缩进的地方了,报错告诉我sess已经关闭。。。

猜你喜欢

转载自blog.csdn.net/dlz_yhn/article/details/129428759