Add command line option for rnn_type and update readme

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

tasks/copy_task.py

@@ -24,6 +24,7 @@ from dnc.dnc import DNC
 
 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')