TensorFlow SavedModel保存和加载模型

最新推荐文章于 2024-03-15 23:48:28 发布

hustqb

最新推荐文章于 2024-03-15 23:48:28 发布

阅读量2.5w

点赞数 5

分类专栏： TensorFlow

本文链接：https://blog.csdn.net/hustqb/article/details/80782143

版权

本文详细介绍了如何使用tf.train.Saver在TensorFlow中保存和加载模型，包括初始化saver、保存模型和变量、加载模型以及加载变量的步骤。通过设置max_to_keep和keep_checkpoint_every_n_hours参数来管理保存的断点文件。此外，还提供了如何使用加载的模型进行fine-tune的示例，特别是针对VGG模型。

摘要由CSDN通过智能技术生成

声明：

参考文献：
版本：Tensorflow 1.1

利用tf.train.Saver()保存和加载模型

"""保存模型和变量"""
v1 = tf.Variable([0], name='v1')
v2 = tf.Variable([0], name='v2')

saver = tf.train.Saver()  # 1. 初始化saver

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    saver.save<