专栏算法工具链在进行模型检测的时候,出现 Pad not support this mode=b'edge'

在进行模型检测的时候,出现 Pad not support this mode=b'edge'

已解决
DuQun2023-04-08
17
4
1.芯片型号:X3派
2.天工开物开发包OpenExplorer版本:XJ3_OE_2.2.2
3.问题定位:模型转换
4.问题具体描述

运行命令:

model_type="onnx"

proto="./onnx/RetinexNet.onnx"

caffe_model="./onnx/RetinexNet.onnx"

output="./log/lenet_checker.log"

march="bernoulli2"

hb_mapper checker --model-type ${model_type} \

--proto ${proto} --model ${caffe_model} \

--march ${march} \

--output ${output} \

--input-shape 1x3x400x600

出现错误:

请问是不支持一些网络结点吗?改进方法是什么呢?

我看算子是支持Pad的

我代码这样写有问题吗?

请求解答

附件:
算法工具链
评论2
0/1000
  • 颜值即正义
    Lv.2

    你好,pad的mode需要是constant,看了你提供的onnx,发现不满足这个要求哈:

    建议看一下源码,是哪些位置产生的这些pad算子,修改一下他们的padding_mode,比如代码中 nn.Conv2D 不要配置padding_mode,直接使用默认值“zeros”

    2023-04-10
    0
    2
    • DuQun回复颜值即正义:

      好的,感谢,昨天看算子列表的时候,我也发现了这个问题,感谢感谢!!!

      2023-04-10
      0
    • 颜值即正义回复DuQun:

      不客气哈,解决了就好

      2023-04-10
      1
  • 颜值即正义
    Lv.2
    2023-04-24
    0
    0