Bug小能手系列(python)_6:‘Upsample‘ object has no attribute ‘recompute_scale_factor‘

最新推荐文章于 2025-03-21 07:30:00 发布

sjx_alo

最新推荐文章于 2025-03-21 07:30:00 发布

阅读量2.8k

点赞数 7

分类专栏： Bug小能手文章标签： python bug pytorch YOLO

本文链接：https://blog.youkuaiyun.com/qq_36758270/article/details/130630794

版权

Bug小能手专栏收录该内容

15 篇文章

订阅专栏

该错误发生在高版本的Pytorch中，原因是使用了YOLOv5代码时版本不匹配。解决方案是创建新的环境并安装Pytorch1.9.0版本，提供了针对不同平台（如ROCM和CUDA）的安装命令。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

‘Upsample’ object has no attribute ‘recompute_scale_factor’

错误

在当前普遍的Pytorch版本上使用YOLO v5的代码，在运行过程出会出现上述错误。

报错。提示

 File "C:\Users\14541\.conda\envs\torch_gpu\lib\site-packages\torch\nn\modules\upsampling.py", line 154, in forward
    recompute_scale_factor=self.recompute_scale_factor)
  File "C:\Users\14541\.conda\envs\torch_gpu\lib\site-packages\torch\nn\modules\module.py", line 1207, in __getattr__
    raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'

解决方案

首先分析错误原因：
该错误的主要原因是因为版本不匹配，当前使用的Pytorch版本过高。
由于该函数对Pytorch整体框架影响程度不大，因此存在两种解决方案。

直接进入Pytorch源码对应的报错的文件（本文中的位置在envs\torch_gpu\lib\site-packages\torch\nn\modules\upsampling.py中），然后将对应的recompute_scale_factor=self.recompute_scale_factor部分删除就行。这里需要注意的是：修改的是Pytorch中的源码部分，因此不建议进行这样修改。
创建新的环境，新环境的Pytorch版本建议为：1.9.0。
新环境使用pip安装的代码如下：（如果需要用conda安装可以从官网上查看）

# ROCM 4.2 (Linux only)
pip install torch==1.9.0+rocm4.2 torchvision==0.10.0+rocm4.2 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

# ROCM 4.1 (Linux only)
pip install torch==1.9.0+rocm4.1 torchvision==0.10.0+rocm4.1 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

# ROCM 4.0.1 (Linux only)
pip install torch==1.9.0+rocm4.0.1 torchvision==0.10.0+rocm4.0.1 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 11.1
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 10.2
pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.9.0+cpu torchvision==0.10.0+cpu torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html