解决“TypeError: can‘t convert cuda:0 device type tensor to numpy. ......”问题
问题描述在训练Pytorch模型的时候,报错Traceback(mostrecentcalllast):File"train.py",line469,intrain(hyp,tb_writer,opt,device)File"train.py",line347,intrainsave_dir=log_dir)File"/home/xxx/Detection/test.py",line176,int