我试图使用inspect_checkpoint.py上的代码检查检查点.但是,我无法让它工作,因为它们并没有真正提供一个例子.我尝试过我认为最简单的方法:
tf.python.tools.inspect_checkpoint.print_tensors_in_checkpoint_file(file_name='./tmp/mdl_ckpt',tensor_name='',all_tensors='')
但是我得到的python
没有属性tools
:
AttributeError: module 'tensorflow.python' has no attribute 'tools'
这似乎是一个(令人尴尬的)琐碎的错误/问题.有人知道发生了什么吗?为什么找不到工具呢?此外,即使它确实找到了,如何运行该文件中提供的功能?
不幸的是,这个非常相关的问题并没有真正解决如何解决这个问题.问题在这里如何找到保存在tensorflow检查点中的变量名?
试试这个:
from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file print_tensors_in_checkpoint_file(file_name='./tmp/mdl_ckpt', tensor_name='', all_tensors=False)
all_tensors
自Tensorflow 0.12.0-rc0以来添加了该参数.