From aae6d260c7f989bd75c2667c833f88bbf5113b68 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Thu, 14 Jan 2021 10:03:09 +0100
Subject: [PATCH] number of agents (fixed or 1..)
---
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 340a21f..2d513ed 100755
--- a/reinforcement_learning/multi_agent_training.py
+++ b/reinforcement_learning/multi_agent_training.py
@@ -510,7 +510,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=5000, type=int)
- parser.add_argument("--n_agent_fixed", help="hold the number of agent fixed", default=False, type=bool)
+ parser.add_argument("--n_agent_fixed", help="hold the number of agent fixed", action='store_true')
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