From 6a2fd3820040f7c8d0b321bebabc4d5e818c9410 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Fri, 8 Jan 2021 09:39:25 +0100
Subject: [PATCH] Tensorboard support added
---
reinforcement_learning/multi_agent_training.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/reinforcement_learning/multi_agent_training.py b/reinforcement_learning/multi_agent_training.py
index b92362c..ec49640 100755
--- a/reinforcement_learning/multi_agent_training.py
+++ b/reinforcement_learning/multi_agent_training.py
@@ -504,7 +504,7 @@ def eval_policy(env, tree_observation, policy, train_params, obs_params):
if __name__ == "__main__":
parser = ArgumentParser()
- parser.add_argument("-n", "--n_episodes", help="number of episodes to run", default=12000, type=int)
+ parser.add_argument("-n", "--n_episodes", help="number of episodes to run", default=5000, type=int)
parser.add_argument("-t", "--training_env_config", help="training config id (eg 0 for Test_0)", default=1,
type=int)
parser.add_argument("-e", "--evaluation_env_config", help="evaluation config id (eg 0 for Test_0)", default=1,
--
GitLab