From 14893138e13dcad4b7285e66f96796d1ee00fca1 Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Thu, 13 Jun 2019 09:46:33 +0200
Subject: [PATCH] post-merge fix
---
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 0c4ed85..1c747c2 100644
--- a/torch_training/training_navigation.py
+++ b/torch_training/training_navigation.py
@@ -77,7 +77,7 @@ action_prob = [0] * action_size
agent_obs = [None] * env.get_num_agents()
agent_next_obs = [None] * env.get_num_agents()
agent = Agent(state_size, action_size, "FC", 0)
-agent.qnetwork_local.load_state_dict(torch.load('./Nets/avoid_checkpoint15000.pth'))
+agent.qnetwork_local.load_state_dict(torch.load(os.path.join(__file_dirname__, 'Nets', 'avoid_checkpoint15000.pth')))
demo = True
--
GitLab