出现 AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 的解决方法

目录

  • 1. 问题所示
  • 2. 原理分析
  • 3. 解决方法

1. 问题所示

基于YOLOv5网络模型来训练自身的数据集,出现如下问题,截图如下所示:
出现 AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 的解决方法_第1张图片
通过代码bug一层一层往下,主要的问题来源这里:

  File "/home/kkjz/anaconda3/envs/py17/lib/python3.7/site-packages/torch/nn/modules/upsampling.py", line 157, in forward
    recompute_scale_factor=self.recompute_scale_factor)
  File "/home/kkjz/an

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