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

Traceback (most recent call last):
File “E:\yolo\yolov5-5.0\yolov5-5.0\detect.py”, line 72, in detect
pred = model(img, augment=opt.augment)[0]
File “D:\python\anaconda\lib\site-packages\torch\nn\modules\module.py”, line 1110, in _call_impl
return forward_call(*input, **kwargs)
File “E:\yolo\yolov5-5.0\yolov5-5.0\models\yolo.py”, line 123, in forward
return self.forward_once(x, profile) # single-scale inference, train
File “E:\yolo\yolov5-5.0\yolov5-5.0\models\yolo.py”, line 139, in forward_once
x = m(x) # run
File “D:\python\anaconda\lib\site-packages\torch\nn\modules\module.py”, line 1110, in _call_impl
return forward_call(*input, **kwargs)
File “D:\python\anaconda\lib\site-packages\torch\nn\modules\upsampling.py”, line 154, in forward
recompute_scale_factor=self.recompute_scale_factor)
File “D:\python\anaconda\lib\site-packages\torch\nn\modules\module.py”, line 1185, in getattr
raise AttributeError("’{}’ object has no attribute ‘{}’".format(
AttributeError: ‘Upsample’ object has no attribute ‘recompute_scale_factor’

解决办法:

torch版本1.11.0 torchvision版本0.10.2
降低torchvision版本0.10.1
降低torch版本1.9.1

你可能感兴趣的:(深度学习,pytorch,python)