search

longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.54k stars 421 forks source link

Memory increase? #40

Open hyer opened 6 years ago

hyer commented 6 years ago

I compile and run the demo.py success. But when I run the train.py on GPU mode and I found the memory always increase, which make the train process killed. Can any one help. thx!

zackzhao commented 6 years ago

Same issue ,any solution now?

shwangdev commented 6 years ago

Same issue. Any suggestion?

jinyu121 commented 6 years ago

I think it is due to the imap function? I switch to use DataLoader, and seems the memory is OK. Move to #71 for more detail.