yolov5报错：‘Upsample‘ object has no attribute ‘recompute_scale_factor‘

 

  报错内容：

File "/workspace/mydetect.py", line 181, in <module>
    oc.detect_an_image()
  File "/workspace/mydetect.py", line 75, in detect_an_image
    pred = self.model(img, augment=self.opt.augment)[0]
  File "/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
    return forward_call(*input, **kwargs)
  File "/workspace/models/yolo.py", line 117, in forward
    return self.forward_once(x, profile)  # single-scale inference, train
  File "/workspace/models/yolo.py", line 148, in forward_once
    x = m(x)  # run
  File "/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
    return forward_call(*input, **kwargs)
  File "/opt/conda/lib/python3.10/site-packages/torch/nn/modules/upsampling.py", line 157, in forward
    recompute_scale_factor=self.recompute_scale_factor)
  File "/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1269, in __getattr__
    raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'

解决方法：

根据上面的错误提示找到/opt/conda/lib/python3.10/site-packages/torch/nn/modules/upsampling.py的位置，修改upsampling.py代码

注释掉：recompute_scale_factor=self.recompute_scale_factor

修改前：

 修改后：