search

bubbliiiing / mobilenet-yolov4-pytorch

这是一个mobilenet-yolov4的库，把yolov4主干网络修改成了mobilenet，修改了Panet的卷积组成，使参数量大幅度缩小。
MIT License
369 stars 104 forks source link

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

Open summer261 opened 2 years ago

summer261 commented 2 years ago

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

bubbliiiing commented 2 years ago

高特征层显存占用大

summer261 commented 2 years ago

好的，歇歇博主