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

原因

pytorch版本问题

解决办法

顺着路径找到文件upsampling.py并打开,修改里面的代码即可解决, 这个方式不需要重新安装pytorch

YOLOV5 训练好模型测试时出现问题:AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘的解决方法​blog.csdn.net/qq_40280673/article/details/125095353​编辑

找到forward方法, 去掉后传参后面的这句话:

recompute_scale_factor=self.recompute_scale_factor

  • 成功运行

AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘_第1张图片

其他:

 此问题还可发生在

1、模型转onnx的export.py脚本中pytorch版本问题, 模型转换报错

2、另外就是使用到训练好的模型进行推理时候报错。

你可能感兴趣的:(深度学习,机器学习,人工智能)