本文详细介绍了如何在TensorFlow中保存训练好的模型，并演示了如何在后续任务中加载和使用这些模型，以便进行预测或进一步训练。通过实例展示了`tf.train.Saver`和`tf.saved_model` API的用法，帮助读者掌握模型持久化的关键步骤。
（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()
with tf.Session() as sess:
最低0.47元/天 解锁文章
扫一扫
3万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
