From 05f08725a16e01908b19665d15a6aa621462bb95 Mon Sep 17 00:00:00 2001
From: JasonJiazhiZhang <21229070+JasonJiazhiZhang@users.noreply.github.com>
Date: Tue, 11 Jun 2019 15:34:04 -0700
Subject: [PATCH] Expose other measures through get_info in NavRLEnv (#120)
Expose other measures through get_info in NavRLEnv
---
habitat_baselines/train_ppo.py | 7 +------
1 file changed, 1 insertion(+), 6 deletions(-)
diff --git a/habitat_baselines/train_ppo.py b/habitat_baselines/train_ppo.py
index d88edc3d2..3e521a054 100644
--- a/habitat_baselines/train_ppo.py
+++ b/habitat_baselines/train_ppo.py
@@ -86,12 +86,7 @@ class NavRLEnv(habitat.RLEnv):
return done
def get_info(self, observations):
- info = {}
-
- if self.get_done(observations):
- info["spl"] = self.habitat_env.get_metrics()["spl"]
-
- return info
+ return self.habitat_env.get_metrics()
def make_env_fn(config_env, config_baseline, rank):
--
GitLab