From 7365be1ad99ff1a3b07fbdbffea7a84ccd961008 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 5 Jan 2021 16:52:13 +0100
Subject: [PATCH] Policy updated
---
reinforcement_learning/multi_agent_training.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/reinforcement_learning/multi_agent_training.py b/reinforcement_learning/multi_agent_training.py
index 8592a95..b92362c 100755
--- a/reinforcement_learning/multi_agent_training.py
+++ b/reinforcement_learning/multi_agent_training.py
@@ -431,6 +431,7 @@ def train_agent(train_params, train_env_params, eval_env_params, obs_params):
writer.add_scalar("timer/learn", learn_timer.get(), episode_idx)
writer.add_scalar("timer/preproc", preproc_timer.get(), episode_idx)
writer.add_scalar("timer/total", training_timer.get_current(), episode_idx)
+ writer.flush()
def format_action_prob(action_probs):
--
GitLab