司家鑫
1
小白求助,我在使用ai_toolchain上转化一个YOLOV5模型的时候,发现01_check.sh时候发现MaxPool_236 BPU id(0) HzQuantizedMaxPool 正常的映射。但是在03_build.sh时候发现并没有maxpool并没有发现对应的实现,输出如下,不知理解是否有误:
ERROR model conversion failed: [ONNXRuntimeError] : 9 : NOT_IMPLEMENTED : Could not find an implementation for the node MaxPool_236:MaxPool(12)
您这边使用的是我们model_zoo中的模型还是自己训练的呢?自己训练的模型需要进行一定的调整,我们的用户手册有详细说明
司家鑫
4
你好,我参考了用户手册进行了调整,同时按照手册统一了yolov5和pytorch的版本,但是输出的模型依旧报错,内容一样。我对比了贵公司释放的yolov5模型,同样使用了和我一模一样的三个maxpool,但是在第三步build没有报错,比较奇怪,还想咨询下您有什么建议。
我们对模型进行了相应的调整,具体信息可以参照readme文件,里面有具体的操作方式,按照步骤操作即可。