Adding notes on memory requirements? about neuraltalk2 HOT 6 CLOSED

Can you try decreasing the batch size to 1, and then decreasing rnn hidden size to say, 128? 2GB is really very little.

from neuraltalk2.

The CPU error is because you're using my pretrained checkpoint, which was trained on GPU. Try running with CPU from scratch, should work ok.

from neuraltalk2.

adding -batch_size 1 fixed it! thanks a bunch!

yes, i just noticed the CPU/GPU thing. i tried moving the require out of the if but that just created another problem, apparently it's not that simple :)

from neuraltalk2.

ok, you also want to make batch size as large as you can fit, by the way, and you should expect to probably decrease the learning rate (if training).

from neuraltalk2.

i don't understand how you have time to be so helpful & encouraging, and work on a phd at the same time. thank you so much! :)

edit: for the record, -batch_size 8 is the highest i can go with the net you posted.

from neuraltalk2.

I was trying to train CPU-based model and encountered same OOM error on Ubuntu. I tried adjusting batch_size and It perfectly works right now. Thanks for the detailed explanation!

from neuraltalk2.