从检查点将训练过的权重注入TensorFlow模型而无需还原

Question

如何将来自检查点文件的训练后的权重注入TensorFlow模型？

我有一个训练有素的模型，但是当我尝试恢复它并以通常的方式运行它时，我会感到恐惧

  

FailedPreconditionError（请参阅上面的回溯）：尝试使用   未初始化

但是，使用checkpoint inspector，我可以看到所有正确的权重都在检查点中。我无法返回并使用其他保存逻辑重新运行模型，因此我想直接从检查点注入权重。

# This does not work.
# It is a method in my model object def load(self, job_name):
def load(self, job_name, sess=None):
        if not sess:
            raise AttributeError("TensorFlow session not provided.")
        saver = tf.train.Saver(self.vars)
        out_fname = OUT_DIR + job_name + "/" + "w.ckpt"
        #
        save_path = saver.save(sess, out_fname)
        saver.restore(sess, save_path)