如何从Tensorflow中的检查点文件加载单个张量?

时间:2018-03-16 16:57:28

标签: python tensorflow

我正在尝试加载存储在已保存检查点文件中的张量。检查点以“V2格式”保存。 示例:“checkpoint-100000.data-00000-of-00001”和“checkpoint-100000.index”。 没有.meta文件

在线可用的所有资源都指向.meta文件作为加载到会话中的基本文件,但我没有选择它。

我只是希望能够查看张量的维度,并对它们进行一些分析。例如,当我运行“print_tensors_in_checkpoint_file()”时,我的一个较大的张量有一个省略号,所以我丢失了很多列/信息,我想恢复原始大小。

带有椭圆的大张量的片段:

Uri.EscapeDataString()

1 个答案:

答案 0 :(得分:0)

使用tf.train.load_variable,例如

x=tf.train.load_variable('/home/user/logdir/checkpoint', 'variable_name')

也很有用:您可以使用tf.train.train.list_variables()

在检查点中获取变量的名称。