#save to file import tensorflow as tf import numpy as np ##(1)Save to file 把相关变量存储到文件中 #remember to define the same dtype and shape when restore W = tf.Variable([[1,2,3],[3,4,5]],dtype=tf.float32,name=‘weights‘) b = tf.Variable([[1,2,3]],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,"my_net/save_net.ckpt") print("Save to path : ",save_path) ##(2)restore variables 从文件中取出相关变量 #redefine the same shape and same type for you variables W = tf.Variable(np.arange(6).reshape((2,3)),dtype=tf.float32,name="weights")#reshape((2,3):2行3列 b = tf.Variable(np.arange(3).reshape((1,3)),dtype=tf.float32,name="biases") #not need init step saver = tf.train.Saver() with tf.Session() as sess: saver.restore(sess,"my_net/save_net.ckpt") print("weights: ",sess.run(W)) print("biases: ",sess.run(b))
原文地址:https://www.cnblogs.com/Harriett-Lin/p/9594132.html
时间: 2024-10-22 08:48:27