onnx转ncnn后的depthwise层无法加载,显示出现nan

k75qkfdt  于 3个月前  发布在  其他
关注(0)|答案(1)|浏览(40)

error log | 日志或报错信息 | ログ

使用convertmodel.com直接转换的模型。不带任何优化操作。最后nncn加载报错 layer load_model 45 Conv_50 failed ,模型中出现nan
这个模型最原始作者提供的onnx模型,

model | 模型 | モデル

  1. original model
    原始onnx模型

how to reproduce | 复现步骤 | 再現方法

1.使用 convertmodel.com 进行转换

2.shion使用load_model 加载模型

ipakzgxi

ipakzgxi1#

针对onnx模型转换的各种问题,推荐使用最新的pnnx工具转换到ncnn
In view of various problems in onnx model conversion, it is recommended to use the latest pnnx tool to convert your model to ncnn

pip install pnnx
pnnx model.onnx inputshape=[1,3,224,224]

详细参考文档
Detailed reference documentation
https://github.com/pnnx/pnnx
https://github.com/Tencent/ncnn/wiki/use-ncnn-with-pytorch-or-onnx#how-to-use-pnnx

相关问题