search

alexgkendall / SegNet-Tutorial

Files for a tutorial to train SegNet for road scenes using the CamVid dataset
http://mi.eng.cam.ac.uk/projects/segnet/tutorial.html
847 stars 517 forks source link

Out of memory #26

Closed rshanor closed 8 years ago

rshanor commented 8 years ago

Hi,

I am trying to work my way through the SegNet tutorial. During the training step, I right away get the error 'error == cudaSuccess (2 vs. 0) out of memory'. I am trying to train on my 4GB GTX980. I am have tried both SegNet and SegNet basic, with batch size as small as just 1. Is there anything else that may be causing this error? Or is my graphics card just insufficient.

Thanks, Rick

mrgloom commented 8 years ago

Same https://github.com/alexgkendall/caffe-segnet/issues/21

rshanor commented 8 years ago

I ended up having to downsample the images by half, and can now run with batch size 7.

mrgloom commented 8 years ago

Can you post resulting *.prototxt file?

My file is here: https://gist.github.com/mrgloom/f0972272938adfc44163

I tried to reduce dimensions of images but still have errors.

rshanor commented 8 years ago

Can you repost your errors? I will try and help you debug. All I changed were upsample parameters, as well as the prediction outputs.