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

显示全部楼层 · 发表于 2020-3-17 15:04:51

pytorch版本？
log？

显示全部楼层 · 发表于 2020-3-17 15:20:04

问题已提交

显示全部楼层 · 发表于 2020-3-18 20:29:55

请将模型或下载地址、转换脚本发给我

显示全部楼层 · 发表于 2020-3-20 16:38:25

NPU部门答复：
densenet201的模型规模比较大，我这边转换都要花20多分钟，netron也打不开，通过分析，发现这个模型的concat的input数量（47个）已经超过了vip8000的硬件限制（16个），因此该模型在vip8000上暂时无法支持。

显示全部楼层 · 发表于 2020-5-9 15:56:22

18022443868 发表于 2020-5-9 15:35
Vip8000是什么？RK3399pRO上面的硬件吗？我的模型中有41个concat，那也同样不支持，这样理解没错吧？这个 ...

NPU的硬件。
简单这么说，这个网络我们硬件限制，支持不了。

显示全部楼层 · 发表于 2020-5-9 21:13:06

18022443868 发表于 2020-5-9 16:59
那如果说放在RK1808计算棒呢？是否能够支持？我在PC端插上计算棒也是同样的错误，估计也支持不了 ...

你的网络我们NPU支持不了，你说的所有rknn目前都是这款NPU，无法支持

显示全部楼层 · 发表于 2020-5-11 10:02:53

18022443868 发表于 2020-5-11 09:58
那我将concat 输入数量降低到16以下，RKNN是否就可以支持我的网络了？

动手试试

显示全部楼层 · 发表于 2020-5-19 11:44:48

18022443868 发表于 2020-5-19 11:38
可以，但是我加入out = F.interpolate(input,size=(skip.size(2), skip.size(3)),mode="bilinear",align_ ...

加载失败一般都是模型转的有问题，看下转换过程有没有报错，再用UI工具看下转换完的rknn模型是不是正确

显示全部楼层 · 发表于 2020-5-19 12:00:08

18022443868 发表于 2020-5-19 11:58
转换过程未出错。应该是F.interpolate操作有点特殊，需要对应怎么去处理之类的？ ...

pytorch我只能给NPU team报bug。。。
pytorch现在自定义op是否支持我也没关注，如果是tf或者caffe，一般做法就是自定义op自己写.
你这里的 F.interpolate 是对应nn. 里的哪个函数？

显示全部楼层 · 发表于 2020-5-19 14:36:28

18022443868 发表于 2020-5-19 14:09
torch.nn.functional.interpolate

已经报给NPU部门了