Toybrick

楼主: luxiaofei

pytorch一个模型转化过程中的报错

jefferyzhang

版主

积分
13003
楼主
发表于 2020-3-12 17:20:42 | 显示全部楼层
问题我也上报给NPU部门了,方便的话可以提供下模型+转换脚本,便于他们复现问题
回复

使用道具 举报

jefferyzhang

版主

积分
13003
沙发
发表于 2020-3-12 17:37:10 | 显示全部楼层
luxiaofei 发表于 2020-3-12 17:26
好的,模型大于10Mb了论坛好像不能上传,有邮箱吗?我把模型和代码发到您邮箱里? ...

发百度网盘给链接就可以。
回复

使用道具 举报

jefferyzhang

版主

积分
13003
板凳
发表于 2020-3-17 11:10:07 | 显示全部楼层
问题将在1.4.0版本中解决,1.3.0中不支持conv2d的weight为另外一个卷积的输出。
回复

使用道具 举报

jefferyzhang

版主

积分
13003
地板
发表于 2020-3-17 14:52:19 | 显示全部楼层
luxiaofei 发表于 2020-3-17 14:17
1.4.0版本大概什么时候推出啊老哥?

不知道,不要问我- -# 我不是NPU部门的
回复

使用道具 举报

jefferyzhang

版主

积分
13003
5#
发表于 2020-3-24 14:20:08 | 显示全部楼层
Sean 发表于 2020-3-24 13:48
老哥这句话是什么意思啊,我可以绕过这个问题吗

就是说1.3.0还不支持这个做法,1.4.0已经支持,但是要等我们发布
回复

使用道具 举报

jefferyzhang

版主

积分
13003
6#
发表于 2020-3-24 16:50:18 | 显示全部楼层
Sean 发表于 2020-3-24 16:38
1.3.0是可以支持torch的上采样吧,我现在定位到上采样这层报错,如果去掉上采样,模型就能转换成功 ...

转换成功还要试下运行结果是否与PC一致。。
上采样总类很多,没问NPU部门我也不太懂pytorch支持情况怎么样。
不过现在很多bug都报过去了,相信1.4.0出来pytorch的支持会更好
回复

使用道具 举报

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

本版积分规则

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


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