GPU-Accelerated Deep Learning Library in Python
Hebel is a library for deep learning with neural networks in Python using GPU acceleration with CUDA through PyCUDA. It implements the most important types of neural network models and offers a variety of different activation functions and training methods such as momentum, Nesterov momentum, dropout, and early stopping.
Currently, Hebel will only run on Linux and probably Mac OS X (not tested). Hebel currently won't run in Windows, because scikits.cuda is not supported in Windows.
- PyCUDA
- scikits.cuda
- numpy
- PyYAML
- skdata (only for MNIST example)
Study the yaml configuration files in examples/
and run
python train_model.py examples/mnist_neural_net_shallow.yaml
The script will create a directory in examples/mnist
where the models and logs are saved.
Hebel is the German word for lever, one of the oldest tools that humans use. As Archimedes said it: "Give me a lever long enough and a fulcrum on which to place it, and I shall move the world."