原因
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
- 成功运行
其他:
此问题还可发生在
1、模型转onnx的export.py脚本中pytorch版本问题, 模型转换报错
2、另外就是使用到训练好的模型进行推理时候报错。