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

Published 20 hours ago • verified publisher icon longcw

Run this code will occur memory leak?

Open mshmoon opened this issue 7 years ago • 1 comments

Every time I run this code some steps,my computer will crash. My computer have 8g memory ,gtx1060 NVIDIA. I set batch_size=1 I use 'top' command to monitor the usage of memory ,the memory and nvidia memory continuously reduce.

mshmoon avatar Apr 18 '18 11:04 mshmoon

This may caused by multiprocessing.Pool.imap. You can use the dataloader from https://github.com/longcw/yolo2-pytorch/issues/71

longcw avatar May 06 '18 12:05 longcw