reinforcement_learning fix reward threshold

lijielife · Aug 15, 2017 · 2dca104 · 2dca104
1 parent 86bc3e5
commit 2dca104
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/reinforcement_learning/actor_critic.py b/reinforcement_learning/actor_critic.py
@@ -99,7 +99,7 @@ def finish_episode():
     if i_episode % args.log_interval == 0:
         print('Episode {}\tLast length: {:5d}\tAverage length: {:.2f}'.format(
             i_episode, t, running_reward))
-    if running_reward > 200:
+    if running_reward > env.spec.reward_threshold:
         print("Solved! Running reward is now {} and "
               "the last episode runs to {} time steps!".format(running_reward, t))
         break
diff --git a/reinforcement_learning/reinforce.py b/reinforcement_learning/reinforce.py
@@ -89,7 +89,7 @@ def finish_episode():
     if i_episode % args.log_interval == 0:
         print('Episode {}\tLast length: {:5d}\tAverage length: {:.2f}'.format(
             i_episode, t, running_reward))
-    if running_reward > 200:
+    if running_reward > env.spec.reward_threshold:
         print("Solved! Running reward is now {} and "
               "the last episode runs to {} time steps!".format(running_reward, t))
         break