jingtianyilong / YOLOv4-pytorch

This is a pytorch repository of YOLOv4 that can be trained with custom dataset.
31 stars 12 forks source link

CUDA OUT OF MEMORY #12

Open songeza opened 3 years ago

songeza commented 3 years ago

I tried many methods, such as adjusting batch size, but still oom. In fact, my machine memory is 32g and my video memory is 8g. Batch should not appear batch size cannot be set more than 8. And bacth_size is reduced, oom will also appear after running several epochs. How to solve it?