diff --git a/reinforcement_learning/multi_agent_training.py b/reinforcement_learning/multi_agent_training.py index 8592a95eb78359fa3e3c82e7e9110573d76ff606..b92362ced5de725a4c2988227c84b3c9160e0137 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):