From da9d805a9e3050b836774fc9cbd7d3e573c0185b Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Wed, 10 Jul 2019 16:24:28 +0200
Subject: [PATCH] #42 run baselines in ci
---
torch_training/bla.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/torch_training/bla.py b/torch_training/bla.py
index cd1aa13..698b4fe 100644
--- a/torch_training/bla.py
+++ b/torch_training/bla.py
@@ -82,6 +82,12 @@ def main(argv):
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_checkpoint30000.pth") as file_in:
+ agent.qnetwork_local.load_state_dict(torch.load(file_in))
+
+ demo = False
+ record_images = False
+ frame_step = 0
print("multi_agent_trainging.py (2)")
--
GitLab