ncnn
ncnn copied to clipboard
permute(2, 0, 3, 1, 4):onnx2ncnn不支持perm[0]==0的情况
qkv = (self.qkv(x).reshape(B, N, 3, self.num_heads, C // self.num_heads).permute(2, 0, 3, 1, 4).contiguous()) 我看onnx2ncnn.cpp的代码里不支持perm[0]==0的情况,请问一下这种perm[0]也变化的情况要怎么处理。
针对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