From 99ff2dc828f050a3da656c5f47deecf960cb800d Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Mon, 15 Jul 2019 10:39:05 -0400
Subject: [PATCH] minor update
---
torch_training/render_agent_behavior.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/torch_training/render_agent_behavior.py b/torch_training/render_agent_behavior.py
index f98318e..1d760c6 100644
--- a/torch_training/render_agent_behavior.py
+++ b/torch_training/render_agent_behavior.py
@@ -100,7 +100,7 @@ for trials in range(1, n_trials + 1):
# Run episode
for step in range(max_steps):
- env_renderer.render_env(show=True, show_observations=False, show_predictions=True)
+ env_renderer.render_env(show=True, show_observations=False, show_predictions=False)
if record_images:
env_renderer.gl.saveImage("./Images/flatland_frame_{:04d}.bmp".format(frame_step))
--
GitLab