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

显示全部楼层 · 发表于 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:30:21

问题已经上报

显示全部楼层 · 发表于 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必须是固定的不能是变化的