【Bug解决】TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy...
问题分析笔者在训练yolov5模型时出现如下问题:Traceback(mostrecentcalllast):File"train.py",line460,intrain(hyp,opt,device,tb_writer)File"train.py",line320,intrainplots=epoch==0orfinal_epoch)#plotfirstandlastFile"D:\yolov5