Closed shiyutang closed 1 year ago
可以在转换命令里加上 -debug 打印出详细日志看一下
增加选项之后日志如下:mobilenetv3.log
我发现增加了pooling代码均会存在,转tnn时与上面相同的错误:D/tnn: get_node_attr_ai [File /opt/TNN/tpython3: /opt/TNN/tools/onnx2tnn/src/core/layer/onnx_converter_pool.cc:132: virtual std::__cxx11::string OnnxOpConverterPool::TNNLayerParam(onnx::NodeProto&, OnnxNetInfo&): Assertion `0' failed. 查看代码之后发现是pooling的pads.size落在不支持的范畴:
目前转模型会错误的代码示例如下,不知道能否如何修改pool参数,使其能成功被tnn支持?
可以把 AdaptiveAvgPool 换成 AvgPool,同时显式的设置 kernel size 和 padding 看看
issue 长时间没有回复,先关闭了,如果有需要的话可以再打开
1. 使用环境(environment)
OS Version: 7
2. Github版本
commit(optional):
3. 详细描述bug 情况 (Describe the bug) A clear and concise description of what the bug is.
original model:PaddlePaddle -》 onnx (如果可以的话,请上传原始的模型文件) 模型文件:https://github.com/shiyutang/files/raw/main/mobileseg_mobilenetv3_ade20k_512x512_160k_model.onnx
4. 日志(Log)
5. 截图(Screenshots)