Skip to content

S
seminar-breakout
Commit a62ce63a authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

bug fix in limitlength

parent 8e5bcbe6
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tensorpack/RL/common.py
View file @ a62ce63a
...@@ -54,9 +54,10 @@ class LimitLengthPlayer(ProxyPlayer): ...@@ -54,9 +54,10 @@ class LimitLengthPlayer(ProxyPlayer):
self.cnt += 1 self.cnt += 1
if self.cnt >= self.limit: if self.cnt >= self.limit:
isOver = True isOver = True
if isOver:
self.finish_episode() self.finish_episode()
self.restart_episode() self.restart_episode()
if isOver:
self.cnt = 0
return (r, isOver) return (r, isOver)
def restart_episode(self): def restart_episode(self):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment