From 8e474dbddf0710ece2f262aa5984b7ec402e3114 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Fri, 24 May 2019 15:24:25 +0200
Subject: [PATCH] updated other paths
---
torch_training/training_navigation.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/torch_training/training_navigation.py b/torch_training/training_navigation.py
index b30278c..b9f203e 100644
--- a/torch_training/training_navigation.py
+++ b/torch_training/training_navigation.py
@@ -210,5 +210,5 @@ for trials in range(1, n_trials + 1):
eps,
action_prob / np.sum(action_prob)))
torch.save(agent.qnetwork_local.state_dict(),
- '../flatland/baselines/Nets/avoid_checkpoint' + str(trials) + '.pth')
+ './Nets/avoid_checkpoint' + str(trials) + '.pth')
action_prob = [1] * 4
--
GitLab