即使从 ckpt 文件恢复后，张量流权重也不会恢复

Question

我使用以下代码，恢复的权重是随机初始化，而不是存储在ckpt文件中的实际权重。请帮助我了解我哪里出错了。

best_val_model = 'val_E1_A86.ckpt'
model_dir = './models/'
with tf.Session(config = config) as sess:
    sess.run(tf.global_variables_initializer())
    print('Testing the model on 10000 Images!')
    ckpt_file = os.path.join(model_dir, best_val_model)
    saver = tf.train.import_meta_graph(ckpt_file)
    saver.restore(sess, ckpt_file)
    weights = {}
    for v in tf.trainable_variables():
        weights[v.name] = v.eval()

在

训练期间使用 tf.train.Saver(sess,filename) 保存的实际模型权重。在恢复时，随机权重正在恢复。

Answer 1

我只需要删除两行sess.run(tf.golbal_variables_initializer())，saver = tf.train.import_meta_graph(ckpt_file)，它工作正常。使用的最终代码：

saver = tf.train.Saver(tf.trainable_variables())
with tf.Session(config = config) as sess:
   sess.run(tf.local_variables_initializer())
   print('Testing the model on 10000 Images!')
   ckpt_file = os.path.join(model_dir, best_val_model)
   saver.restore(sess, ckpt_file)