tensorflow1.0 模型的保存与加载

最新推荐文章于 2023-01-07 14:27:58 发布

高颜值的杀生丸

最新推荐文章于 2023-01-07 14:27:58 发布

阅读量263

点赞数

本文链接：https://blog.csdn.net/u010970956/article/details/104872807

版权

import tensorflow as tf
import numpy as np

# ##Save to file
# W = tf.Variable([[4,5,6],[7,8,9]],dtype=tf.float32,name="weight")
# b = tf.Variable([[2,5,8]],dtype=tf.float32,name="biases")
#
# init = tf.initialize_all_variables()
#
# saver = tf.train.Saver()
#
# with tf.Session() as sess:
#     sess.run(init)
#     save_path = saver.save(sess,"models/model.ckpt")
#     print("save complete")

W = tf.Variable(np.arange(6).reshape((2,3)),dtype=tf.float32,name="weight")
b = tf.Variable(np.arange(3).reshape((1,3)),dtype=tf.float32,name="biases")

saver = tf.train.Saver()

with tf.Session() as sess:
    saver.restore(sess,"models/model.ckpt")
    print(sess.run(W))
    print(sess.run(b))

[[4. 5. 6.]
[7. 8. 9.]]
[[2. 5. 8.]]