ethantang95 · ethantang95 · Jun 23, 2017 · Oct 18, 2016 · Oct 21, 2016 · Oct 28, 2016
diff --git a/.gitignore b/.gitignore
@@ -17,3 +17,13 @@ TAGS
 /build/
 /dist/
 *.egg-info/
+
+#Intellij files
+.idea/
+
+#vscode
+.vscode/
+
+#.project
+.project
+/.project
diff --git a/.travis.yml b/.travis.yml
@@ -1,4 +1,4 @@
-# Copyright (c) 2015-2016, NVIDIA CORPORATION.  All rights reserved.
+# Copyright (c) 2015-2017, NVIDIA CORPORATION.  All rights reserved.
 
 os: linux
 dist: trusty
@@ -10,7 +10,6 @@ env:
     global:
         - CAFFE_ROOT=~/caffe
         - TORCH_ROOT=~/torch
-        # Fixes for Torch and OpenBLAS
         - OMP_NUM_THREADS=1
         - OPENBLAS_MAIN_FREE=1
         - secure: "WSqrE+PQm76DdoRLRGKTK6fRWfXZjIb0BWCZm3IgHgFO7OE6fcK2tBnpDNNw4XQjmo27FFWlEhxN32g18P84n5PvErHaH65IuS9Nv6FkLlPXZlVqGNxbPmEA4oTkD/6Y6kZyZWZtLh2+/1ijuzQAPnIy/4BEuL8pdO+PsoJ9hYM="
@@ -20,6 +19,7 @@ env:
         - DIGITS_TEST_FRAMEWORK=torch
         - DIGITS_TEST_FRAMEWORK=tensorflow
         - DIGITS_TEST_FRAMEWORK=none
+        - DIGITS_TEST_FRAMEWORK=none WITH_PLUGINS=false
 
 matrix:
     include:
@@ -43,6 +43,7 @@ matrix:
                       - dput
                       - gnupg
           install:
+              - git fetch --unshallow
               - git remote add nvidia-digits-upstream https://github.com/NVIDIA/DIGITS.git  # for forks
               - git fetch nvidia-digits-upstream --tags
               - pip install twine
@@ -130,13 +131,11 @@ install:
     - echo "backend:agg" > ~/.config/matplotlib/matplotlibrc
     - ./scripts/travis/install-caffe.sh $CAFFE_ROOT
     - if [ "$DIGITS_TEST_FRAMEWORK" == "torch" ]; then travis_wait ./scripts/travis/install-torch.sh $TORCH_ROOT; else unset TORCH_ROOT; fi
+    - pip install -r ./requirements.txt --force-reinstall
     - if [ "$DIGITS_TEST_FRAMEWORK" == "tensorflow" ]; then travis_wait ./scripts/travis/install-tensorflow.sh; fi
-    - pip install -r ./requirements.txt
     - pip install -r ./requirements_test.txt
     - pip install -e .
-    - pip install -e ./plugins/data/imageGradients
-    - pip install -e ./plugins/view/imageGradients
+    - if [ "$WITH_PLUGINS" != "false" ]; then find ./plugins/*/* -maxdepth 0 -type d | xargs -n1 pip install -e; fi
 
 script:
     - ./digits-test -v
-
diff --git a/LICENSE b/LICENSE
@@ -1,4 +1,4 @@
-Copyright (c) 2014-2016, NVIDIA CORPORATION. All rights reserved.
+Copyright (c) 2014-2017, NVIDIA CORPORATION. All rights reserved.
 
 Redistribution and use in source and binary forms, with or without
 modification, are permitted provided that the following conditions

diff --git a/README.md b/README.md
@@ -4,11 +4,13 @@
 
 DIGITS (the **D**eep Learning **G**PU **T**raining **S**ystem) is a webapp for training deep learning models.
 
+The currently supported frameworks are: Caffe 1, Torch, and Tensorflow
+
 # Installation
 
 | Installation method | Supported platform[s] | Available versions | Instructions |
 | --- | --- | --- | --- |
-| Deb packages | Ubuntu 14.04 | [14.04 repo](http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64) | [docs/UbuntuInstall.md](docs/UbuntuInstall.md) |
+| Deb packages | Ubuntu 14.04, 16.04 | [14.04 repo](http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64), [16.04 repo](http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64) | [docs/UbuntuInstall.md](docs/UbuntuInstall.md) |
 | Docker | Linux | [DockerHub tags](https://hub.docker.com/r/nvidia/digits/tags/) | [nvidia-docker wiki](https://github.com/NVIDIA/nvidia-docker/wiki/DIGITS) |
 | Source | Ubuntu 14.04, 16.04 | [GitHub tags](https://github.com/NVIDIA/DIGITS/releases) | [docs/BuildDigits.md](docs/BuildDigits.md) |
 
@@ -18,6 +20,7 @@ Once you have installed DIGITS, visit [docs/GettingStarted.md](docs/GettingStart
 
 Then, take a look at some of the other documentation at [docs/](docs/) and [examples/](examples/):
 
+* [Getting started with TensorFlow](docs/GettingStartedTensorflow.md)
 * [Getting started with Torch](docs/GettingStartedTorch.md)
 * [Fine-tune a pretrained model](examples/fine-tuning/README.md)
 * [Train an autoencoder network](examples/autoencoder/README.md)

diff --git a/digits-devserver b/digits-devserver
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright (c) 2016, NVIDIA CORPORATION.  All rights reserved.
+# Copyright (c) 2016-2017, NVIDIA CORPORATION.  All rights reserved.
 
 set -e
 

diff --git a/digits-lint b/digits-lint
@@ -1,13 +1,13 @@
 #!/bin/bash
-# Copyright (c) 2016, NVIDIA CORPORATION.  All rights reserved.
+# Copyright (c) 2016-2017, NVIDIA CORPORATION.  All rights reserved.
 
 set -e
 
 echo "=== Checking for Python lint ..."
 if which flake8 >/dev/null 2>&1; then
-    python2 `which flake8` .
+    python2 `which flake8` --exclude ./examples,./digits/standard-networks/tensorflow .
 else
-    python2 -m flake8 .
+    python2 -m flake8 --exclude ./examples,./digits/standard-networks/tensorflow .
 fi
 
 echo "=== Checking for JavaScript lint ..."

diff --git a/digits-test b/digits-test
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright (c) 2014-2016, NVIDIA CORPORATION.  All rights reserved.
+# Copyright (c) 2014-2017, NVIDIA CORPORATION.  All rights reserved.
 
 set -e
 

diff --git a/digits-walkthrough b/digits-walkthrough
diff --git a/digits/__init__.py b/digits/__init__.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2014-2016, NVIDIA CORPORATION.  All rights reserved.
+# Copyright (c) 2014-2017, NVIDIA CORPORATION.  All rights reserved.
 from __future__ import absolute_import
 
 from .version import __version__

diff --git a/digits/__main__.py b/digits/__main__.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2014-2016, NVIDIA CORPORATION.  All rights reserved.
+# Copyright (c) 2014-2017, NVIDIA CORPORATION.  All rights reserved.
 
 import argparse
 import os.path

diff --git a/digits/config/__init__.py b/digits/config/__init__.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2015-2016, NVIDIA CORPORATION.  All rights reserved.
+# Copyright (c) 2015-2017, NVIDIA CORPORATION.  All rights reserved.
 from __future__ import absolute_import
 
 # Create this object before importing the following imports, since they edit the list

diff --git a/digits/config/caffe.py b/digits/config/caffe.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2015-2016, NVIDIA CORPORATION.  All rights reserved.
+# Copyright (c) 2015-2017, NVIDIA CORPORATION.  All rights reserved.
 from __future__ import absolute_import
 
 import imp

diff --git a/digits/config/gpu_list.py b/digits/config/gpu_list.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2015-2016, NVIDIA CORPORATION.  All rights reserved.
+# Copyright (c) 2015-2017, NVIDIA CORPORATION.  All rights reserved.
 from __future__ import absolute_import
 
 from . import option_list

diff --git a/digits/config/jobs_dir.py b/digits/config/jobs_dir.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2015-2016, NVIDIA CORPORATION.  All rights reserved.
+# Copyright (c) 2015-2017, NVIDIA CORPORATION.  All rights reserved.
 from __future__ import absolute_import
 
 import os