最近在看MKR的算法,复现跑通了以后,把训练好的模型保存下来
加入以下代码即可
saver = tf.train.Saver()
saver.save(sess, '../model/MKRmodel.ckpt')
print('模型已保存')
上面的代码要写到和sess.run()一同缩进的地方,不然会报错。
中间代码太长了,第一次写到sess.run()前面那个缩进的地方了,报错告诉我sess已经关闭。。。
最近在看MKR的算法,复现跑通了以后,把训练好的模型保存下来
加入以下代码即可
saver = tf.train.Saver()
saver.save(sess, '../model/MKRmodel.ckpt')
print('模型已保存')
上面的代码要写到和sess.run()一同缩进的地方,不然会报错。
中间代码太长了,第一次写到sess.run()前面那个缩进的地方了,报错告诉我sess已经关闭。。。