From 599d89f92fd058505cc5d250cef63b1776030a85 Mon Sep 17 00:00:00 2001
From: Unknown <aravindbk92@gmail.com>
Date: Wed, 6 Feb 2019 21:48:29 -0500
Subject: [PATCH] success check for high level policy
---
backends/kerasrl_learner.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/backends/kerasrl_learner.py b/backends/kerasrl_learner.py
index 642554c..a341711 100644
--- a/backends/kerasrl_learner.py
+++ b/backends/kerasrl_learner.py
@@ -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))
--
GitLab