背景
运行python语言实现的深度学习模型【选用pytorch】代码时，代码报错
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'方案
将报错文件“D:\Anaconda3\envs\yolov5-6.0\lib\site-packages\torch\nn\modules\upsampling.py”【具体路径请查看报错信息】大体类似
里边的forward方法修改一下，改成以下
def forward(self, input: Tensor) -> Tensor:
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)修改后程序运行正常
文章出处登录后可见！
已经登录？立即刷新
