From 1dd88043f1383def066bf6d1a8c82a86bf44609a Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sun, 1 Sep 2019 09:33:47 -0400
Subject: [PATCH] updated training and inference to flatland 2.0 level
generators
---
torch_training/multi_agent_inference.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/torch_training/multi_agent_inference.py b/torch_training/multi_agent_inference.py
index 8c1cbd0..2fbbe61 100644
--- a/torch_training/multi_agent_inference.py
+++ b/torch_training/multi_agent_inference.py
@@ -88,7 +88,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)
-with path(torch_training.Nets, "avoid_checkpoint60000.pth") as file_in:
+with path(torch_training.Nets, "avoid_checkpoint100.pth") as file_in:
agent.qnetwork_local.load_state_dict(torch.load(file_in))
record_images = False
--
GitLab