版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Dian1pei2xiao3/article/details/82290675
import tensorflow as tf
import numpy as np
##Save to file
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.tables_initializer()
with tf.Session()as sess:
sess.run(init)
saver=tf.train.Saver()
save_path=saver.save(sess,"my_net/save_net.ckpt")
print("Save to path:",save_path)