Toybrick

pytorch转换模型出错

jefferyzhang

版主

积分
13003
楼主
发表于 2020-4-29 21:02:19 | 显示全部楼层
本帖最后由 jefferyzhang 于 2020-4-29 21:03 编辑

意思是'aten::unsqueeze
这个op不支持,我会报给相关部门处理。你可以试下1.3.2版本,目前1.3.2版本新增了pytorch很多op的支持
回复

使用道具 举报

jefferyzhang

版主

积分
13003
沙发
发表于 2020-5-8 08:30:08 | 显示全部楼层
NPU Team回复:

这个op前几天已更新,等下一个版本发布就可以使用了。
如果用的是torchvision里的模型,可以这样:models.googlenet(pretrained=True, transform_input=False),把输入预处理放到外面来做,这样就可以避开unsqueeze。
回复

使用道具 举报

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

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表