bug fix in limitlength

a62ce63a · Yuxin Wu · 8e5bcbe6 · a62ce63a
Commit a62ce63a authored Aug 05, 2016 by Yuxin Wu
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tensorpack/RL/common.py tensorpack/RL/common.py +2 -1

No files found.
--- a/tensorpack/RL/common.py
+++ b/tensorpack/RL/common.py
@@ -54,9 +54,10 @@ class LimitLengthPlayer(ProxyPlayer):
        self.cnt += 1
        if self.cnt >= self.limit:
            isOver = True
-        if isOver:
            self.finish_episode()
            self.restart_episode()
+        if isOver:
+            self.cnt = 0
        return (r, isOver)

    def restart_episode(self):