在一个文件中保存变量,打印出的变量值是101,在另一个文件中恢复变量,打印的值是1,是什么原因? 要怎么保存更新后的参数?


在一个文件中保存变量,打印出的变量值是101,在另一个文件中恢复变量,打印的值是1,是什么原因? 要怎么保存更新后的参数?
A文件代码:
import tensorflow as tf
 
W=tf.Variable(1,name='W1')
def w_c(W):
    W=W+1
    return W
init=tf.global_variables_initializer()
saver=tf.train.Saver(tf.global_variables(),max_to_keep=15)
with tf.Session() as sess:
    sess.run(init)
    for i in range(100):
        W=w_c(W)
    a=sess.run(W)
    print(a,flush=True)
    saver.save(sess,'modeltest/savertest.ckpt')


B文件代码:
import tensorflow as tf
W=tf.Variable(2,name='W1')
new_saver = tf.train.Saver()
with tf.Session() as sess:
    new_saver.restore(sess, 'modeltest/savertest.ckpt')
    print(sess.run(W))
已邀请:

要回复问题请先登录注册

返回顶部