From 4af2cc0110ec2ae849cd5445614155250bc4cbff Mon Sep 17 00:00:00 2001
From: Erik Nygren <baerenjesus@gmail.com>
Date: Tue, 9 Jun 2020 09:34:20 +0000
Subject: [PATCH] Update training_navigation.py
---
torch_training/training_navigation.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/torch_training/training_navigation.py b/torch_training/training_navigation.py
index fe4905f..e03c5d1 100644
--- a/torch_training/training_navigation.py
+++ b/torch_training/training_navigation.py
@@ -42,10 +42,11 @@ def main(argv):
# Use a the malfunction generator to break agents from time to time
- stochastic_data = {'malfunction_rate': 8000, # Rate of malfunction occurence of single agent
- 'min_duration': 15, # Minimal duration of malfunction
- 'max_duration': 50 # Max duration of malfunction
- }
+ stochastic_data = MalfunctionParameters(malfunction_rate=10000, # Rate of malfunction occurence
+ min_duration=15, # Minimal duration of malfunction
+ max_duration=50 # Max duration of malfunction
+ )
+
# Custom observation builder
TreeObservation = TreeObsForRailEnv(max_depth=2)
@@ -69,7 +70,8 @@ def main(argv):
malfunction_generator_and_process_data=malfunction_from_params(stochastic_data),
# Malfunction data generator
obs_builder_object=TreeObservation)
-
+ # Reset env
+ env.reset(True,True)
# After training we want to render the results so we also load a renderer
env_renderer = RenderTool(env, gl="PILSVG", )
# Given the depth of the tree observation and the number of features per node we get the following state_size
--
GitLab