人工智能从入门到精通（10）

最新推荐文章于 2024-09-24 09:53:21 发布

为不为

最新推荐文章于 2024-09-24 09:53:21 发布

阅读量306

点赞数

分类专栏：人工智能文章标签：人工智能

本文链接：https://blog.csdn.net/u011747152/article/details/78670661

版权

人工智能专栏收录该内容

17 篇文章 0 订阅

订阅专栏

tensorflow模型持久化

为了可以让训练结果可以复用

import tensorflow as tf

v1=tf.Variable(tf.constant(1.0,shape=[1]),name="v1")
v2=tf.Variable(tf.constant(2.0,shape=[1]),name="v2")
result=v1+v2
init_op=tf.initialize_all_variables()
saver=tf.train.Saver()
with tf.Session() as sess:
    sess.run(init_op)
    saver.save(sess,"/path/to/model/model.ckpt")

读取保存的数据

import tensorflow as tf

saver=tf.train.import_meta_graph("/path/to/model/model.ckpt.meta")
with tf.Session() as sess:
    saver.restore(sess,"/path/to/model/model.ckpt")
    print(sess.run(tf.get_default_graph().get_tensor_by_name("add:0")))