dotfold/digits

Deep Learning GPU Training System

DIGITS

DIGITS (the Deep Learning GPU Training System) is a webapp for training deep learning models. The currently supported frameworks are: Caffe, Torch, and Tensorflow.

Installation

Installation method	Supported platform[s]	Available versions	Instructions
Docker	Linux	DockerHub tags	nvidia-docker wiki
Source	Ubuntu 14.04, 16.04	GitHub tags	docs/BuildDigits.md

Usage

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
• 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
• Use Python layers in your Caffe networks
• Download a model and use it to classify an image outside of DIGITS
• Overview of the REST API

Get help

Installation issues

• First, check out the instructions above
• Then, ask questions on our user group

Usage questions

• First, check out the Getting Started page
• Then, ask questions on our user group

Bugs and feature requests

• Please let us know by filing a new issue
• Bonus points if you want to contribute by opening a pull request!
  • You will need to send a signed copy of the Contributor License Agreement to [email protected] before your change can be accepted.