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

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的上采样操作？

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

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

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

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

问题已经上报

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

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

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

好的，收到，等贵司回复

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

目前用的模型是FcHarDNet

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必须是固定的不能是变化的