Tensorflow如何检查模型

时间:2019-02-22 19:42:36

标签: tensorflow machine-learning output

我建立了一个Tensorflow RNN模型并想检查模型结果(例如,使用了哪些功能/变量以及使用程度如何等)

我创建了以下文件:

  • 检查点
  • my_time_series_model
  • my_time_series_model.index
  • my_time_series_model元数据
  • train.PBTXT

但是我在读取这些文件时遇到了问题。我发现以下代码:

from tensorflow.python import pywrap_tensorflow

model_file = "/trained/checkpoint"
reader = pywrap_tensorflow.NewCheckpointReader(model_file)
var_to_shape_map = reader.get_variable_to_shape_map()

for key in sorted(var_to_shape_map):
    print("tensor_name: ", key)
    print(reader.get_tensor(key))

我收到以下错误:

  

检查点:数据丢失:不是稳定的(错误的魔术数字):也许是   文件格式不同,您需要使用其他文件格式   恢复运算符?

检查点文件是否错误?它在我的文件夹中没有扩展名,只是说类型是数据?

任何帮助都会很棒!

1 个答案:

答案 0 :(得分:0)

您可以使用tensorflow python工具中的inspect_checkpoint函数检查检查点的张量。

张量流文档中的示例:

# import the inspect_checkpoint library
from tensorflow.python.tools import inspect_checkpoint as chkp

# print all tensors in checkpoint file
chkp.print_tensors_in_checkpoint_file("/tmp/model.ckpt", tensor_name='', all_tensors=True)

# tensor_name:  v1
# [ 1.  1.  1.]
# tensor_name:  v2
# [-1. -1. -1. -1. -1.]

https://www.tensorflow.org/guide/saved_model#inspect_variables_in_a_checkpoint