Skip to content
Snippets Groups Projects
Commit ebd2868b authored by Erik Nygren's avatar Erik Nygren
Browse files

updated paths in training_navigation.py

parent c7d40119
No related branches found
No related tags found
No related merge requests found
...@@ -46,7 +46,7 @@ env = RailEnv(width=20, ...@@ -46,7 +46,7 @@ env = RailEnv(width=20,
number_of_agents=3) number_of_agents=3)
""" """
env_renderer = RenderTool(env, gl="QT") env_renderer = RenderTool(env, gl="QTSVG")
handle = env.get_agent_handles() handle = env.get_agent_handles()
state_size = 105 * 2 state_size = 105 * 2
...@@ -66,7 +66,7 @@ action_prob = [0] * 4 ...@@ -66,7 +66,7 @@ action_prob = [0] * 4
agent_obs = [None] * env.get_num_agents() agent_obs = [None] * env.get_num_agents()
agent_next_obs = [None] * env.get_num_agents() agent_next_obs = [None] * env.get_num_agents()
agent = Agent(state_size, action_size, "FC", 0) 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 demo = True
...@@ -144,7 +144,6 @@ for trials in range(1, n_trials + 1): ...@@ -144,7 +144,6 @@ for trials in range(1, n_trials + 1):
for step in range(100): for step in range(100):
if demo: if demo:
env_renderer.renderEnv(show=True) env_renderer.renderEnv(show=True)
time.sleep(0.1)
# print(step) # print(step)
# Action # Action
for a in range(env.get_num_agents()): for a in range(env.get_num_agents()):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment