Add command line option for rnn_type and update readme

ixaxaar · Oct 27, 2017 · 89298d3 · 89298d3
1 parent e2a007e
commit 89298d3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -59,8 +59,9 @@ output, (controller_hidden, memory, read_vectors) = \
 
 The copy task, as descibed in the original paper, is included in the repo.
 
+From the project root:
 ```
-python ./copy_task.py -cuda 0
+python ./tasks/copy_task.py -cuda 0
 ```
 
 ## General noteworthy stuff

diff --git a/tasks/copy_task.py b/tasks/copy_task.py
@@ -24,6 +24,7 @@
 
 parser = argparse.ArgumentParser(description='PyTorch Differentiable Neural Computer')
 parser.add_argument('-input_size', type=int, default= 6, help='dimension of input feature')
+parser.add_argument('-rnn_type', type=str, default='lstm', help='type of recurrent cells to use for the controller')
 parser.add_argument('-nhid', type=int, default=64, help='humber of hidden units of the inner nn')
 
 parser.add_argument('-nlayer', type=int, default=2, help='number of layers')