yolov5——bug总结

AttributeError

一、‘Upsample’ object has no attribute ‘recompute_scale_factor’

进入upsampling.py文件，找到对应的forward（）函数

直接把最后一个属性recompute_scale_factor删除即可

---

二、‘Hardswish’ object has no attribute ‘inplace’

进入activation.py文件，找到forward（）函数，将reuturn语句中的self.inplace删除即可

TypeError

can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

进入报错的文件 _tensor.py，找到 array(self, dtype=None)函数

将两个return语句后面的self.numpy() 改成 self.cpu().numpy()即可

因为GPU中的tensor得先放在CPU中才能转换成numpy数据，因为numpy是cpu only。