Running Densnet in CPU - Githubissues

titu1994 / DenseNet

DenseNet implementation in Keras

MIT License

707 stars 294 forks source link

Running Densnet in CPU #48

Closed saisivasriram closed 5 years ago

saisivasriram commented 5 years ago

I ran Densenet.py for mnist dataset, But training got hanged with CPU utilization in 1st epoch itself, Kindly suggest a way to run densenet keras for mnist dataset, and advise of the procedure to run the same in GPU.

titu1994 commented 5 years ago

DenseNets are very memory exhaustive and time consuming models. It is not possible to train them on CPU for now. On GPU, there is no change to this repo.