From e3ead28c3bcfcb82ca757d0b549cf82462d90bbd Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sun, 27 Oct 2019 22:51:43 -0400
Subject: [PATCH] updated training file to max time and added multi speed and
stochasticity
---
torch_training/multi_agent_training.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index 0e928d0..2ceb3e1 100644
--- a/torch_training/multi_agent_training.py
+++ b/torch_training/multi_agent_training.py
@@ -168,8 +168,8 @@ def main(argv):
# Collection information about training
tasks_finished = 0
- for agent in env.agents:
- if agent.status == RailAgentStatus.DONE_REMOVED:
+ for current_agent in env.agents:
+ if current_agent.status == RailAgentStatus.DONE_REMOVED:
tasks_finished += 1
done_window.append(tasks_finished / max(1, env.get_num_agents()))
scores_window.append(score / max_steps) # save most recent score
--
GitLab