search

bodokaiser / piwise

Pixel-wise segmentation on VOC2012 dataset using pytorch.
BSD 3-Clause "New" or "Revised" License
383 stars 86 forks source link

Out of Memory Issue #16

Open zy937085318 opened 6 years ago

zy937085318 commented 6 years ago

Do any one meet the error that: RuntimeError: cuda runtime error (2) : out of memory at /opt/conda/conda-bld/pytorch_1511320797808/work/torch/lib/THC/generic/THCStorage.cu:66 I just simply apply this model on titanX and also on my own 1080ti, same issue appeared.

NicWickman commented 6 years ago

Lower your batch size to 1 and see if it persists, you're probably just running out of VRAM on the hungry networks.

fantasyzl commented 6 years ago

How long did it take to run the program?