scaelles / DEXTR-PyTorch

Deep Extreme Cut http://www.vision.ee.ethz.ch/~cvlsegmentation/dextr

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

out of memory

CJ0109 opened this issue · comments

when I train PASCAL dataset on GTX 750ti, I encountered the "cuda error(2), out of memory", is this normal?

We trained on a Titan-X, aka 12 GB memory GPU.
If your memory is not sufficient, try decreasing the batch size.