search

argusswift / YOLOv4-pytorch

This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO
1.68k stars 329 forks source link

CUDA OUT OF MEMORY #222

Closed songeza closed 2 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 64g and my video memory is 11g. Batch should not appear batch size cannot be set more than 16. And bacth_size is reduced, oom will also appear after running several epochs. How to solve it?

EdwardLocke commented 2 years ago

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

Hello, I also encountered the same problem, have you solved it?

songeza commented 2 years ago

Not yet.

Get Outlook for Androidhttps://aka.ms/AAb9ysg

From: EdwardLocke @.> Sent: Friday, December 10, 2021 9:44:23 PM To: argusswift/YOLOv4-pytorch @.> Cc: feng-ai @.>; Author @.> Subject: Re: [argusswift/YOLOv4-pytorch] CUDA OUT OF MEMORY (#222)

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

Hello, I also encountered the same problem, have you solved it?

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHubhttps://github.com/argusswift/YOLOv4-pytorch/issues/222#issuecomment-990985722, or unsubscribehttps://github.com/notifications/unsubscribe-auth/AQOTVUCXN5RNVMGS3LKGH23UQH73PANCNFSM5B5JUSMQ. Triage notifications on the go with GitHub Mobile for iOShttps://apps.apple.com/app/apple-store/id1477376905?ct=notification-email&mt=8&pt=524675 or Androidhttps://play.google.com/store/apps/details?id=com.github.android&referrer=utm_campaign%3Dnotification-email%26utm_medium%3Demail%26utm_source%3Dgithub.