Firefly开源社区

打印 上一主题 下一主题

[人工智能] rknn不支持torch.nn.ZeroPad2d

26

积分

0

威望

0

贡献

技术小白

积分
26

rknn不支持torch.nn.ZeroPad2d

发表于 2022-8-16 11:54:52      浏览:4963 | 回复:0        打印      只看该作者   [复制链接] 楼主
问题描述及复现步骤:
pytorch模型中如果有torch.nn.ZeroPad2d((0,1,0,1)),转换模型时报错:_p_fuse_pad_into_maxpool
去掉这部分就能转换成功

firefly.zip

348 Bytes, 下载次数: 0, 下载积分: 灯泡 -1 , 经验 -1

回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

友情链接 : 爱板网 电子发烧友论坛 云汉电子社区 粤ICP备14022046号-2
快速回复 返回顶部 返回列表