Skip to content

groupnorm在c++上调用出错 #4310

@MrCary

Description

@MrCary

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

image

context | 编译/运行环境 | バックグラウンド

window10
torchversion: '1.12.1+cu116'
pnnx:20221020-windows
ncnn:22.10.25更新的最新版

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

1.pytorch代码生成pt:相关代码在以下链接
image
2.pnnx转ncnn:
pnnx wav2vec3.pt inputshape=[1,1,400]
3.c++调用报错:
image

more | 其他 | その他

经过实验,nn.groupnorm当输入的是二维(b, c, h,w)时,转成ncnn后,调用不会报错,而当nn.groupnorm当输入的是(b,c,h)时,转成ncnn后,调用就会出错。

相关param及bin链接,及相关代码
链接: https://pan.baidu.com/s/1cw6kYo3SqRiP4M77j5kIhg?pwd=1111 提取码: 1111

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions