Merge pull request #10 from ixaxaar/hidden_layers

Test travis build

README.md

@@ -1,5 +1,7 @@
 # Differentiable Neural Computer, for Pytorch
 
+[![Build Status](https://travis-ci.org/ixaxaar/pytorch-dnc.svg?branch=master)](https://travis-ci.org/ixaxaar/pytorch-dnc) [![PyPI version](https://badge.fury.io/py/dnc.svg)](https://badge.fury.io/py/dnc)
+
 This is an implementation of [Differentiable Neural Computers](http://people.idsia.ch/~rupesh/rnnsymposium2016/slides/graves.pdf), described in the paper [Hybrid computing using a neural network with dynamic external memory, Graves et al.](https://www.nature.com/articles/nature20101)
 
 ## Install

test/test_gru.py

@@ -15,10 +15,9 @@ import sys
 import os
 import math
 import time
-sys.path.append('./src/')
-sys.path.insert(0, os.path.join('..', '..'))
+sys.path.insert(0, '.')
 
-from dnc.dnc import DNC
+from dnc import DNC
 from test_utils import generate_data, criterion
 
 

test/test_lstm.py

@@ -15,10 +15,9 @@ import sys
 import os
 import math
 import time
-sys.path.append('./src/')
-sys.path.insert(0, os.path.join('..', '..'))
+sys.path.insert(0, '.')
 
-from dnc.dnc import DNC
+from dnc import DNC
 from test_utils import generate_data, criterion
 
 

test/test_rnn.py

@@ -15,10 +15,9 @@ import sys
 import os
 import math
 import time
-sys.path.append('./src/')
-sys.path.insert(0, os.path.join('..', '..'))
+sys.path.insert(0, '.')
 
-from dnc.dnc import DNC
+from dnc import DNC
 from test_utils import generate_data, criterion