Fix broken RNN:resetStates()

1 files changed, 4 insertions, 0 deletions

diff --git a/RNN.lua b/RNN.lua
index 80ee760..8b1a965 100644
--- a/RNN.lua
+++ b/RNN.lua
@@ -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