From 08c6d12c340730629df1c6957c73591870958bf6 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Wed, 15 May 2019 13:24:01 +0200
Subject: [PATCH] update requirements_dev.txt
---
examples/training_navigation.py | 8 ++++----
requirements_dev.txt | 1 +
2 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/examples/training_navigation.py b/examples/training_navigation.py
index b60468a..0262640 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 40a6b7f..51687a0 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
--
GitLab