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

‘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整体框架影响程度不大，因此存在两种解决方案。

1. 直接进入Pytorch源码对应的报错的文件（本文中的位置在envs\torch_gpu\lib\site-packages\torch\nn\modules\upsampling.py中），然后将对应的recompute_scale_factor=self.recompute_scale_factor部分删除就行。这里需要注意的是：修改的是Pytorch中的源码部分，因此不建议进行这样修改。
2. 创建新的环境，新环境的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