OpenDIGITS (the Deep Learning GPU Training System) is a webapp for training deep learning models. The currently supported frameworks are: Torch and Tensorflow on Python 3. Keras and Jupyter Notebook support is coming soon.
The primary difference between this and the NVIDIA version is this is designed to be easier to install with python 3 support (pip install soon!) and no old Caffe left-overs
| Installation method | Supported platform[s] | Available versions | Instructions |
|---|---|---|---|
| Source | Ubuntu 14.04, 16.04 | GitHub tags | docs/BuildDigits.md |
| PIP Wheel | OS X, Windows 64 | Coming Soon! | In Preparation |
Once you have installed DIGITS, visit docs/GettingStarted.md for an introductory walkthrough.
Then, take a look at some of the other documentation at docs/ and examples/:
- Getting started with TensorFlow
- Getting started with Torch
- Fine-tune a pretrained model
- Creating a dataset using data from S3 endpoint
- Train an autoencoder network
- Train a regression network
- Train a Siamese network
- Train a text classification network
- Train an object detection network
- Learn more about weight initialization
- Download a model and use it to classify an image outside of DIGITS
- Overview of the REST API
- First, check out the instructions above
- Please let us know by filing a new issue
- Bonus points if you want to contribute by opening a pull request!