gpt4 book ai didi

python - 针对 tensorflow 诗人中其他架构对评估.py 进行的修改

转载 作者:行者123 更新时间:2023-12-01 09:19:25 24 4
gpt4 key购买 nike

有一个用于测量模型性能的脚本 tensorflow-for-poets-2/scripts/evaluate.py 。该脚本最初是为移动网络模型编写的,但可以针对其他架构进行相应修改。我应该在 41 行中写什么

logits = graph.get_tensor_by_name("final_training_ops/Wx_plus_b/add:0"). 

我收到`KeyError:

The name final_training_ops/Wx_plus_b/add:0' refers to a Tensor which does not exist. The operation, 'final_training_ops/Wx_plus_b/add', does not exist in the graph.

最佳答案

这是我自己想出来的。给定 .pb 文件,我们可以使用此 code 查看卡住的 .pb TensorFlow 图中的操作。因此,我们需要替换行 41根据我们的 .pb 文件进行最后一次操作。

关于python - 针对 tensorflow 诗人中其他架构对评估.py 进行的修改,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50921711/

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