/keras_experiments

Playing around with the keras neural networks lib

Primary LanguageJupyter NotebookMIT LicenseMIT

Keras experiments

Experiments done using the keras neural networks lib.

Python requirements listed in the requirements.txt file. And if you want to run it using the gpu (provided you have support for it and the required dependencies), you can do it with:

THEANO_FLAGS="mode=FAST_RUN,device=gpu,floatX=float32,cuda.root=/usr/local/cuda/" ipython notebook

Otherwise you can just run it in the cpu, like any normal ipython notebook.

A test_gpu.py script is provided to validate if you are running it under the cpu or gpu. Run it in that same way to check if your gpu is working. (script copied from the Theano docs)