From 5802f6dced9b9fc665d5975d3ccda1987cb9d55f Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Thu, 11 Jul 2019 08:33:29 +0200
Subject: [PATCH] #42 run baselines in ci
---
torch_training/bla.py | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/torch_training/bla.py b/torch_training/bla.py
index 80ec308..ec81697 100644
--- a/torch_training/bla.py
+++ b/torch_training/bla.py
@@ -110,13 +110,13 @@ def main(argv):
max_steps = int(3 * (env.height + env.width))
agent_obs = [None] * env.get_num_agents()
agent_next_obs = [None] * env.get_num_agents()
- # # Reset environment
- # if file_load:
- # obs = env.reset(False, False)
- # else:
- # obs = env.reset(True, True)
- # if demo:
- # env_renderer.set_new_rail()
+ # Reset environment
+ if file_load:
+ obs = env.reset(False, False)
+ else:
+ obs = env.reset(True, True)
+ if demo:
+ env_renderer.set_new_rail()
# obs_original = obs.copy()
# final_obs = obs.copy()
# final_obs_next = obs.copy()
--
GitLab