赞
踩
错误提示
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'
点击这个错误提示到
修改如下:
def forward(self, input: Tensor) -> Tensor:
# return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners,
# recompute_scale_factor=self.recompute_scale_factor)
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)
注释掉,增加一行,删除不兼容的参数即可。
出现这个问题的原因应该是安装的pytorch版本不兼容导致。
这里的方法是实用性解决。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。