From ebd2868b2cb27ecd2343b95801ec2339de6f6f06 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Thu, 23 May 2019 13:45:37 +0200
Subject: [PATCH] updated paths in training_navigation.py
---
examples/training_navigation.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/examples/training_navigation.py b/examples/training_navigation.py
index 497db9ca..0cb9d275 100644
--- a/examples/training_navigation.py
+++ b/examples/training_navigation.py
@@ -46,7 +46,7 @@ env = RailEnv(width=20,
number_of_agents=3)
"""
-env_renderer = RenderTool(env, gl="QT")
+env_renderer = RenderTool(env, gl="QTSVG")
handle = env.get_agent_handles()
state_size = 105 * 2
@@ -66,7 +66,7 @@ action_prob = [0] * 4
agent_obs = [None] * env.get_num_agents()
agent_next_obs = [None] * env.get_num_agents()
agent = Agent(state_size, action_size, "FC", 0)
-agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint15000.pth'))
+agent.qnetwork_local.load_state_dict(torch.load('./flatland/baselines/Nets/avoid_checkpoint15000.pth'))
demo = True
@@ -144,7 +144,6 @@ for trials in range(1, n_trials + 1):
for step in range(100):
if demo:
env_renderer.renderEnv(show=True)
- time.sleep(0.1)
# print(step)
# Action
for a in range(env.get_num_agents()):
--
GitLab