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

Training in CPU only mode very slow #72

Open sachinroyxavier opened 7 years ago

sachinroyxavier commented 7 years ago

Hi all, I am using caffe in CPU only mode and made the changes in _solver.prototxt. For 80 iterations i was taking abou 6 hours. I was not able to finish the training of segnet, basic as we ĺl as with the pretrained encoder. My CPU infos are i7 quad core@ 2.90Ghz and 8Gb RAM.

Am I missing something in the prototxt files or is this the expected behaviour? Thanks in advance for the help.

WeiyiLi commented 7 years ago

I am also new to caffe but as far as I know, it is not strange that CPU takes longer time. Maybe you can reduce the batch_size in training prototxt.

matbs24 commented 6 years ago

Me too. 35 minutes for 20 iterations. So bad. any solutions?