Run this code will occur memory leak?

longcw / yolo2-pytorch

YOLOv2 in PyTorch

1.54k stars 421 forks source link

Run this code will occur memory leak? #77

Open mshmoon opened 6 years ago

mshmoon commented 6 years ago

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.

longcw commented 6 years ago

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