Open dnkscu opened 1 year ago
作者您好 训练一段时间之后出现这样的问题
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() 就是说GPU环境没办法将tensor转换成numpy,需要在哪里添加".cpu()"呢
你可以尝试修改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()])
作者您好 训练一段时间之后出现这样的问题
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() 就是说GPU环境没办法将tensor转换成numpy,需要在哪里添加".cpu()"呢