Skip to content

Commit

Permalink
Update reset_experience in READMe
Browse files Browse the repository at this point in the history
  • Loading branch information
ixaxaar committed Oct 26, 2017
1 parent c16237e commit 978cdbb
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ pip install dnc
| gpu_id | -1 | ID of the GPU, -1 for CPU |
| independent_linears | False | Whether to use independent linear units to derive interface vector |
| share_memory | True | Whether to share memory between controller layers |
| reset_experience | False | Whether to reset memory (This is a parameter for the forward pass) |


Example usage:
Expand All @@ -51,7 +52,7 @@ rnn = DNC(
(controller_hidden, memory, read_vectors) = (None, None, None)

output, (controller_hidden, memory, read_vectors) = \
rnn(torch.randn(10, 4, 64), (controller_hidden, memory, read_vectors))
rnn(torch.randn(10, 4, 64), (controller_hidden, memory, read_vectors, reset_experience=True))
```

## Example copy task
Expand Down

0 comments on commit 978cdbb

Please sign in to comment.