我建立了一个Tensorflow RNN模型并想检查模型结果(例如,使用了哪些功能/变量以及使用程度如何等)
我创建了以下文件:
但是我在读取这些文件时遇到了问题。我发现以下代码:
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))
我收到以下错误:
检查点:数据丢失:不是稳定的(错误的魔术数字):也许是 文件格式不同,您需要使用其他文件格式 恢复运算符?
检查点文件是否错误?它在我的文件夹中没有扩展名,只是说类型是数据?
任何帮助都会很棒!
答案 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