(1)对训练模型的保存,将模型命名为my_net.ckp,并保存在net/文件夹下
saver = tf.train.Saver()
with tf.Session() as sess:
#保存模型
saver.save(sess,'net/my_net.ckpt')
saver.save(sess,'net/my_net.ckpt')
(2)对本地模型进行调用
saver = tf.train.Saver()
&n

本文详细介绍了如何在TensorFlow中保存训练好的模型,并演示了如何在后续任务中加载和使用这些模型,以便进行预测或进一步训练。通过实例展示了`tf.train.Saver`和`tf.saved_model` API的用法,帮助读者掌握模型持久化的关键步骤。

8177

被折叠的 条评论
为什么被折叠?



