search

scaelles / DEXTR-PyTorch

Deep Extreme Cut http://www.vision.ee.ethz.ch/~cvlsegmentation/dextr
GNU General Public License v3.0
843 stars 153 forks source link

out of memory #11

Closed CJ0109 closed 6 years ago

CJ0109 commented 6 years ago

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

kmaninis commented 6 years ago

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