yolov5训练模型中遇到的GPU数据转到CPU的一个问题：‘list‘ object has no attribute ‘cpu‘

问题描述：
训练模型的过程中，出现

这个问题出现在general.py 中，源码为

return np.array(targets)

按照网上的方法，改为

return np.array(targets.cpu())

没有解决：提示下面错误
后来看到一份博客，也遇到类似的问题
解决方法：

return np.array(torch.tensor(targets, device='cpu'))