fixed bug in generation mode of lstm.lua

wongjoel · Jun 28, 2017 · 7bb7bba · 7bb7bba
1 parent acfea5e
commit 7bb7bba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/lstm.lua b/models/lstm.lua
@@ -98,7 +98,7 @@ function makeLSTM()
     local in_pose = {}
     for i = 1, opt.nPast+opt.nFuture do
       local s = self.s[i - 1]
-      if i <= opt.nPast+opt.nFuture then
+      if i <= opt.nPast then
         in_pose[i] = pose[i]:clone()
       else
         in_pose[i] = gen_pose[i-1]:clone()