wzliu
Results
3
comments of
wzliu
> Now that error doesn't exist. However, as timeSteps get larger, a memory error occurs. GPU memory is only 6GB, batch_size is 1, timesteps=32, Still display GPU memory error. @buaa-luzhi...
请问您解决了这个问题了吗 我同样遇到了这样的问题
你可以尝试修改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()])