From aef71e5022f19b9bca43841ad5cb5b34e9e050da Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Mon, 15 Jul 2019 11:42:30 -0400
Subject: [PATCH] updated file for conflict detection training. Used for
introduction to the problem
---
torch_training/multi_agent_training.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index 0862e30..bee0d96 100644
--- a/torch_training/multi_agent_training.py
+++ b/torch_training/multi_agent_training.py
@@ -44,6 +44,8 @@ def main(argv):
print("main2")
# Get an observation builder and predictor
+ # The predictor will always predict the shortest path from the current location of the agent.
+ # This is used to warn for potential conflicts --> Should be enhanced to get better performance!
predictor = ShortestPathPredictorForRailEnv()
observation_helper = TreeObsForRailEnv(max_depth=tree_depth, predictor=predictor)
--
GitLab