tensorflow模型（ckpt文件）加载

最新推荐文章于 2024-06-04 10:08:46 发布

路上的病人

最新推荐文章于 2024-06-04 10:08:46 发布

阅读量3.2k

点赞数

分类专栏： Tensorflow学习之路

本文链接：https://blog.csdn.net/qq_40994943/article/details/84797725

版权

Tensorflow学习之路专栏收录该内容

12 篇文章 3 订阅

订阅专栏

模型的保存可参考上一篇博客[https://blog.csdn.net/qq_40994943/article/details/84797543]

上代码：

# 加载模型（即加载图和参数）

with tf.Session() as sess:
    saver = tf.train.import_meta_graph('./model_test/model.ckpt-300.meta')# 加载图结构
    saver.restore(sess,tf.train.latest_checkpoint('./model_test/'))# 加载参数
    print(sess.run('w1:0'))显示模型里w1的值

结果：
在这里插入图片描述