toolkit1.3.0+UbuntuPC模拟器+Pytorch DenseNet出错

只看该作者 · 发表于 2020-5-19 14:49:09

jefferyzhang 发表于 2020-5-19 12:00
pytorch我只能给NPU team报bug。。。
pytorch现在自定义op是否支持我也没关注，如果是tf或者caffe，一般 ...

试过pytorch中的几个上采样函数:torch.nn.functional.interpolate torch.nn.functional.upsample torch.nn.functional.upsample_bilinear 三种上采样操作，都不行，是不是目前RKNN还未支持pytorch的上采样操作？

只看该作者 · 发表于 2020-5-19 15:02:23

18022443868 发表于 2020-5-19 14:49
试过pytorch中的几个上采样函数:torch.nn.functional.interpolate torch.nn.functional.upsample ...

你是toolkit 1.3.2 么？1.3.2加了大量pytorch op支持了啊

只看该作者 · 发表于 2020-5-19 17:01:32

jefferyzhang 发表于 2020-5-19 15:02
你是toolkit 1.3.2 么？1.3.2加了大量pytorch op支持了啊

是的，API:1.3.2，DRV1.3.0。
我的pt文件上传一下，发麻烦帮忙看下到底是什么地方存在问题？

只看该作者 · 发表于 2020-5-19 17:30:21

问题已经上报

只看该作者 · 发表于 2020-5-19 17:34:15

jefferyzhang 发表于 2020-5-19 15:02
你是toolkit 1.3.2 么？1.3.2加了大量pytorch op支持了啊

后面给的是未超过16个concat的模型，我这个模型初始化老是失败。麻烦帮忙看下，万分感谢

只看该作者 · 发表于 2020-5-19 18:50:20

jefferyzhang 发表于 2020-5-19 17:30
问题已经上报

好的，收到，等贵司回复

只看该作者 · 发表于 2020-5-21 10:49:02

jefferyzhang 发表于 2020-5-19 17:30
问题已经上报

目前用的模型是FcHarDNet

只看该作者 · 发表于 2020-5-25 20:07:17

NPU Team答复： 253180

尝试将out = F.interpolate(input,size=(skip.size(2), skip.size(3)),mode="bilinear",align_corners=True,)
改成out = F.interpolate(input,size=(int(skip.size(2)), int(skip.size(3))),mode="bilinear",align_corners=True,)
上采样的size必须是固定的不能是变化的

toolkit1.3.0+UbuntuPC模拟器+Pytorch DenseNet出错

本帖子中包含更多资源

本帖子中包含更多资源