专栏算法工具链关于ai_toolchain的问题

关于ai_toolchain的问题

已解决
司家鑫2021-07-09
266
4

小白求助，我在使用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)

算法工具链
评论1
0/1000
  • admin
    Lv.1

    您这边使用的是我们model_zoo中的模型还是自己训练的呢？自己训练的模型需要进行一定的调整，我们的用户手册有详细说明

    2021-07-09
    0
    3
    • 司家鑫回复admin:

      好的，我是自己训练的模型，我再确认下，非常感谢

      2021-07-12
      0
    • 司家鑫回复admin:

      你好，我参考了用户手册进行了调整，同时按照手册统一了yolov5和pytorch的版本，但是输出的模型依旧报错，内容一样。我对比了贵公司释放的yolov5模型，同样使用了和我一模一样的三个maxpool，但是在第三步build没有报错，比较奇怪，还想咨询下您有什么建议。

      2021-07-19
      2
    • admin回复司家鑫:

      我们对模型进行了相应的调整，具体信息可以参照readme文件，里面有具体的操作方式，按照步骤操作即可。

      2021-07-19
      2