Skip to content
Snippets Groups Projects
Commit 08c6d12c authored by Erik Nygren's avatar Erik Nygren :bullettrain_front:
Browse files

update requirements_dev.txt

parent 216d1ae6
No related branches found
No related tags found
No related merge requests found
......@@ -30,9 +30,9 @@ env = RailEnv(width=10,
rail_generator=random_rail_generator(cell_type_relative_proportion=transition_probability),
number_of_agents=1)
"""
env = RailEnv(width=50,
height=50,
rail_generator=complex_rail_generator(nr_start_goal=50, min_dist=5, max_dist=99999, seed=0),
env = RailEnv(width=15,
height=15,
rail_generator=complex_rail_generator(nr_start_goal=5, min_dist=5, max_dist=99999, seed=0),
number_of_agents=5)
"""
env = RailEnv(width=20,
......@@ -59,7 +59,7 @@ scores = []
dones_list = []
action_prob = [0] * 4
agent = Agent(state_size, action_size, "FC", 0)
agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint1500.pth'))
agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint15000.pth'))
demo = True
......
......@@ -18,3 +18,4 @@ matplotlib==3.0.2
PyQt5==5.12
Pillow==5.4.1
svgutils==0.3.1
\ No newline at end of file
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