From 54224ea380f75aa62eab448bec9267c6a9d47970 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Thu, 11 Jul 2019 17:18:01 -0400
Subject: [PATCH] updated to incoroporate new loading method
---
torch_training/multi_agent_training.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index c9fe733..5224ab6 100644
--- a/torch_training/multi_agent_training.py
+++ b/torch_training/multi_agent_training.py
@@ -89,7 +89,7 @@ def main(argv):
with path(torch_training.Nets, "avoid_checkpoint30000.pth") as file_in:
agent.qnetwork_local.load_state_dict(torch.load(file_in))
- demo = True
+ demo = False
record_images = False
frame_step = 0
--
GitLab