New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

onnx导出失败 #14

Open

ZHIZIHUABU opened this issue Jul 12, 2024 · 1 comment

ZHIZIHUABU commented Jul 12, 2024

rcs3-yolo模型无法导出onnx，报错信息为F.avg_pool2d()的kernel_size参数类型时tensor，不是int型。查询到SEBlock模块中有一个F.avg_pool2d()算子,kernel_size=input_size(3)。

Owner

mkang315 commented Jul 17, 2024

Please look at the answer from this link Problem with F.avg_pool2d #34743. Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment