diff --git a/examples/training_navigation.py b/examples/training_navigation.py index b60468a51b8b73a4448a57494357328b130bb7b5..02626405160628a432cef3859299853a050431d5 100644 --- a/examples/training_navigation.py +++ b/examples/training_navigation.py @@ -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 diff --git a/requirements_dev.txt b/requirements_dev.txt index 40a6b7f683a9b268f210b48057d0186b4e89b92b..51687a034ae1183ce9c4343930229dac29825eb6 100644 --- a/requirements_dev.txt +++ b/requirements_dev.txt @@ -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