我做了如下修改:1、重新聚类了12个anchor,2、额外增加了一层检测层。那么预训练权重是不是不管用了?我不载入预训练权重开始训练,显示GPU内存不足。我用的是2060S,8G显存。请问这是显卡不够还是代码修改的问题?
RuntimeError: CUDA out of memory. Tried to allocate 86.00 MiB (GPU 0; 8.00 GiB total capacity; 5.75 GiB already allocated; 0 bytes free; 5.80 GiB reserved
in total by PyTorch)
我做了如下修改:1、重新聚类了12个anchor,2、额外增加了一层检测层。那么预训练权重是不是不管用了?我不载入预训练权重开始训练,显示GPU内存不足。我用的是2060S,8G显存。请问这是显卡不够还是代码修改的问题? RuntimeError: CUDA out of memory. Tried to allocate 86.00 MiB (GPU 0; 8.00 GiB total capacity; 5.75 GiB already allocated; 0 bytes free; 5.80 GiB reserved in total by PyTorch)