search

Tencent / ncnn

ncnn is a high-performance neural network inference framework optimized for the mobile platform
Other
20.22k stars 4.15k forks source link

Unsupported reduction axes ! #5367

Open Hassan-Recognito opened 6 months ago

Hassan-Recognito commented 6 months ago

In pytorch

... norm = torch.norm(x, p=2) return x / norm

I converted this model into onnx and tried to convert to ncnn, got error "Unsupported reduction axes !" How to pass model conversion by modify pytorch script? Thanks.

wzyforgit commented 6 months ago

You can use pnnx to convert model instead of onnx. https://github.com/pnnx/pnnx

nihui commented 1 month ago

针对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