已解决YOLOv5训练报错:RuntimeError: Expected all tensors to be on the same device......
这是发生在集成一个yolov5中没有的检测头head的情况下发生的错误,出现的时候是已经训练起来了,在训练结束时发生的报错,下面是我的解决办法。1、问题出现及分析排查改yolov5的网络进行训练时出的报错:RuntimeError:Expectedalltensorstobeonthesamedevice,butfoundatleasttwodevices,cuda:0andcpu!百思不得其解,