Tensorflow:检查点简单加载

时间:2016-06-27 10:21:09

标签: tensorflow

我有一个检查点文件:

checkpoint-20001 checkpoint-20001.meta

如何从此空间中提取变量,而无需加载以前的模型并启动会话等。

我想做点什么

cp = load(checkpoint-20001)
cp.var_a

1 个答案:

答案 0 :(得分:1)

没有记录,但您可以使用类tf.train.NewCheckpointReader从Python检查检查点的内容。

这是一个使用它的测试用例,因此您可以看到该类的工作原理。 https://github.com/tensorflow/tensorflow/blob/861644c0bcae5d56f7b3f439696eefa6df8580ec/tensorflow/python/training/saver_test.py#L1203

由于它不是文档类,因此其API可能在将来发生变化。