AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘

背景

运行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)

修改后程序运行正常