mobilenet-yolov4-pytorch icon indicating copy to clipboard operation
mobilenet-yolov4-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

您好，我用了您的V3+yolov4，我多加了一个检测层104*104*24

Open summer261 opened this issue 3 years ago • 2 comments

我做了如下修改：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)

summer261 avatar Feb 17 '22 15:02 summer261

高特征层显存占用大

bubbliiiing avatar Feb 17 '22 16:02 bubbliiiing

好的，歇歇博主

summer261 avatar Feb 17 '22 16:02 summer261