From 3b5d4040ba51cc1b1b3da32ba4fef0227b65b94c Mon Sep 17 00:00:00 2001
From: "S.P. Mohanty" <spmohanty91@gmail.com>
Date: Thu, 24 Oct 2019 17:46:42 +0200
Subject: [PATCH] Add internal env reset time to stats and use dummyobs in
client.py
---
flatland/evaluators/client.py | 10 +++++++---
1 file changed, 7 insertions(+), 3 deletions(-)
diff --git a/flatland/evaluators/client.py b/flatland/evaluators/client.py
index 9dc1e587..d96e8c5e 100644
--- a/flatland/evaluators/client.py
+++ b/flatland/evaluators/client.py
@@ -225,13 +225,15 @@ class FlatlandRemoteClient(object):
obs_builder_object=obs_builder_object
)
+ time_start = time.time()
local_observation, info = self.env.reset(
regenerate_rail=False,
regenerate_schedule=False,
activate_agents=False,
random_seed=random_seed
)
-
+ time_diff = time.time() - time_start
+ self.update_running_mean_stats("internal_env_reset_time", time_diff)
# Use the local observation
# as the remote server uses a dummy observation builder
return local_observation, info
@@ -296,8 +298,7 @@ if __name__ == "__main__":
_action[_idx] = np.random.randint(0, 5)
return _action
- my_observation_builder = TreeObsForRailEnv(max_depth=3,
- predictor=ShortestPathPredictorForRailEnv())
+ my_observation_builder = DummyObservationBuilder()
episode = 0
obs = True
@@ -318,7 +319,10 @@ if __name__ == "__main__":
while True:
action = my_controller(obs, remote_client.env)
+ time_start = time.time()
observation, all_rewards, done, info = remote_client.env_step(action)
+ time_diff = time.time() - time_start
+ print("Step Time : ", time_diff)
if done['__all__']:
print("Current Episode : ", episode)
print("Episode Done")
--
GitLab