YOLOv4-pytorch icon indicating copy to clipboard operation
YOLOv4-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon jingtianyilong

CUDA OUT OF MEMORY

Open songeza opened this issue 3 years ago • 0 comments

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?

songeza avatar Aug 11 '21 10:08 songeza