YOLOv5报AttributeError:‘Upsample‘objecthasnoattribute‘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

Original: https://blog.csdn.net/weixin_44712708/article/details/123519427
Author: 研究代码的小白
Title: YOLOv5报AttributeError:‘Upsample‘objecthasnoattribute‘recompute_scale_factor‘

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/711572/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球