search

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏
Apache License 2.0
1.5k stars 446 forks source link

yolov4 怎么稀疏训练，一直报错？？ #118

Closed DENESTY closed 3 years ago

DENESTY commented 3 years ago

RuntimeError: CUDA out of memory. Tried to allocate 86.00 MiB (GPU 0; 10.76 GiB total capacity; 9.77 GiB already allocated; 42.25 MiB free; 96.27 MiB cached)