GPU训练下无法使用numpy

[DnkNju]

作者您好
训练一段时间之后出现这样的问题

TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu()
就是说GPU环境没办法将tensor转换成numpy，需要在哪里添加".cpu()"呢

[WuTi0525]

你可以尝试修改utils/utils.py line 819 中的 targets.append([i, cls, x, y, w, h, conf])为 targets.append([i, cls, x.cpu(), y.cpu(), w.cpu(), h.cpu(), conf.cpu()])