Skip to content

Commit 2dca104

Browse files
pbartetsoumith
authored andcommitted
reinforcement_learning fix reward threshold
1 parent 86bc3e5 commit 2dca104

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

reinforcement_learning/actor_critic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def finish_episode():
9999
if i_episode % args.log_interval == 0:
100100
print('Episode {}\tLast length: {:5d}\tAverage length: {:.2f}'.format(
101101
i_episode, t, running_reward))
102-
if running_reward > 200:
102+
if running_reward > env.spec.reward_threshold:
103103
print("Solved! Running reward is now {} and "
104104
"the last episode runs to {} time steps!".format(running_reward, t))
105105
break

reinforcement_learning/reinforce.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def finish_episode():
8989
if i_episode % args.log_interval == 0:
9090
print('Episode {}\tLast length: {:5d}\tAverage length: {:.2f}'.format(
9191
i_episode, t, running_reward))
92-
if running_reward > 200:
92+
if running_reward > env.spec.reward_threshold:
9393
print("Solved! Running reward is now {} and "
9494
"the last episode runs to {} time steps!".format(running_reward, t))
9595
break

0 commit comments

Comments
 (0)