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

yolov5运行detect.py文件出现错误

提示

  File "/home/taobotics/rosenv/lib/python3.7/site-packages/torch/nn/modules/upsampling.py", line 154

去到这个目录修改upsampling.py下的

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)

改为

def forward(self, input: Tensor) -> Tensor:
        return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/SSS__jq/article/details/123458804
————————————————