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

CPU Training #30

Closed aabobakr closed 8 years ago

aabobakr commented 8 years ago

I don't have a powerful GPU for now.. so, Is it possible to train this model on a CPU instead of GPU ?..

I have already tried and it doesn't work.. Am I missing something ?.. or this model doesn't support CPU training ?

alexgkendall commented 8 years ago

You can run the model in CPU mode by selecting solver_mode: CPU in the solver.prototxt