Between Tuesday August 20th, 5:00pm and Thursday August 22nd, 8:00am git.uwaterloo.ca will be down for an upgrade to version 10.8.7.

599d89f9 by Unknown

success check for high level policy

1 parent 788a93ad
......@@ -383,9 +383,10 @@ class DQNLearner(LearnerBase):
termination_reason_counter[termination_reason] += 1
else:
termination_reason_counter[termination_reason] = 1
env.reset()
if episode_reward >= success_reward_threshold:
#TODO: remove below env-specific code
if env.env.goal_achieved:
success_count += 1
env.reset()
print("Episode {}: steps:{}, reward:{}".format(
n + 1, step, episode_reward))
......