gpt4 book ai didi

python - 当 TensorFlow 找不到 tools 属性时,如何检查 TensorFlow 中检查点文件中的变量?

转载 作者:太空狗 更新时间:2023-10-29 20:37:01 25 4
gpt4 key购买 nike

我试图使用 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'

这似乎是一个(令人尴尬的)微不足道的错误/问题。有人知道发生了什么事吗?为什么找不到工具?此外,即使它确实找到了它,如何运行该文件中提供的函数?


不幸的是,这个非常相关的问题并没有真正提供如何解决这个问题的答案。问题在这里How can find the variable names that saved in tensorflow checkpoint?

最佳答案

试试这个:

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)

Tensorflow 0.12.0-rc0 以来添加了 all_tensors 参数.

关于python - 当 TensorFlow 找不到 tools 属性时,如何检查 TensorFlow 中检查点文件中的变量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41867191/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com