From ac9c80165c14332346e7a92500c4ded38f3ede16 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Fri, 3 May 2019 15:35:57 +0200
Subject: [PATCH] first tests with new tree search observation. looks good :)
---
examples/training_navigation.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/examples/training_navigation.py b/examples/training_navigation.py
index 1111e0bb..ddf10b10 100644
--- a/examples/training_navigation.py
+++ b/examples/training_navigation.py
@@ -55,9 +55,9 @@ 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_checkpoint14900.pth'))
+agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint14900.pth'))
-demo = False
+demo = True
def max_lt(seq, val):
--
GitLab