ShichenLiu / CondenseNet

CondenseNet: Light weighted CNN for mobile devices
MIT License
694 stars 131 forks source link

How to run Condensenet without GPU? #34

Closed jakobjackson2016 closed 3 years ago

jakobjackson2016 commented 4 years ago

@ShichenLiu and @gaohuang : I want to run this model on my machine without GPU. How do I exactly do this? What lines of code should I modify, if any?

Command: python main.py --model condensenet -b 64 -j 12 cifar10 --stages 14-14-14 --growth 8-16-32 --gpu 0 --resume requires my machine to have a GPU which my machine doesn’t have.

My goal is to run this model on CPU (machine without GPU). Thank you!