Tensorflow:检查点简单加载

Question

我有一个检查点文件:

checkpoint-20001 checkpoint-20001.meta

我如何从这个空间中提取变量，而不必加载以前的模型和开始会话等

我想做一些像

cp = load(checkpoint-20001)
cp.var_a

Answer 1

没有文档说明，但是您可以使用类tf.train.NewCheckpointReader从Python检查检查点的内容。

下面是一个使用它的测试用例，因此您可以看到该类是如何工作的。https://github.com/tensorflow/tensorflow/blob/861644c0bcae5d56f7b3f439696eefa6df8580ec/tensorflow/python/training/saver_test.py L1203

由于它不是一个文档化的类，它的API将来可能会改变。