diff options
author | Soumith Chintala <soumith@gmail.com> | 2017-03-05 22:06:07 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-03-05 22:06:07 +0300 |
commit | 18510b6482e2202efa6b594de28efd3ea8046892 (patch) | |
tree | 4108b0fb2ac0925c9b8e00e3f75a9ba0dc5d3116 | |
parent | 9c2b867e38a2c75ae610aa312848a3cdb35ae033 (diff) | |
parent | 818bd41611ee80b8a5c7c10948cadb9c1c08c3db (diff) |
Merge pull request #349 from richardassar/resetStates_fix
Fix broken RNN:resetStates()
-rw-r--r-- | RNN.lua | 4 |
1 files changed, 4 insertions, 0 deletions
@@ -235,6 +235,10 @@ function RNN:resetStates() if self.cellInput then self.cellInput = nil end + + self.hiddenOutput:zero() + self.cellOutput:zero() + if self.gradHiddenOutput then self.gradHiddenOutput = nil end |