From 07724835706e9c43484c14e1a817feb29a81a31c Mon Sep 17 00:00:00 2001
From: mlerik <baerenjesus@gmail.com>
Date: Mon, 8 Jul 2019 16:53:16 +0000
Subject: [PATCH] Update Getting_Started_Training.md
---
torch_training/Getting_Started_Training.md | 13 +++++++++++++
1 file changed, 13 insertions(+)
diff --git a/torch_training/Getting_Started_Training.md b/torch_training/Getting_Started_Training.md
index 2734041..b4a5a6f 100644
--- a/torch_training/Getting_Started_Training.md
+++ b/torch_training/Getting_Started_Training.md
@@ -50,3 +50,16 @@ Each node is filled with information gathered along the path to the node. Curren
- 9: agent in the opposite direction
- n = number of agents present other direction than myself
- 0 = no agent present other direction than myself
+
+For training purposes the tree is flattend into a single array.
+
+## Training
+
+Let us now train a simle double dueling DQN agent to navigate to its target on flatland. We start by importing flatland
+```
+from flatland.envs.generators import complex_rail_generator
+from flatland.envs.observations import TreeObsForRailEnv
+from flatland.envs.rail_env import RailEnv
+from flatland.utils.rendertools import RenderTool
+from utils.observation_utils import norm_obs_clip, split_tree
+```
\ No newline at end of file
--
GitLab