pb模型查找不到ArgMax：0张量

来源：12-9 模型测试

戈好雨

2020-10-22

老师好。测试的时候报错pb模型里面没有名为ArgMax的张量，我把所有张量都打印出来确实没有，连最早的Softmax张量都不见了。是不是Pb模型封装错了？

写回答

1回答

会写代码的好厨师

2020-10-29

已采纳

模型是你自己训练生成的么？最后一个节点是什么？属性模型应该就是分类或者回归的模型，分类模型的话如果不需要置信度，不需要接入softmax，直接算argmax就可以了。可以把最后一个节点发给我看一下。

戈好雨

非常感谢！

2020-11-04

共1条回复

Python3+TensorFlow打造人脸识别智能小程序

理论与实战项目双管齐下，让AI技术真正落地应用，适合毕设展示。

1095 学习 · 538 问题

相似问题

回答 1

回答 1

回答 1

回答 1

回答 1