From 6155bff892656263e46277a0f5bfd5fcef741508 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Mon, 18 Jan 2021 16:49:33 +0100
Subject: [PATCH] eps disabled
---
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 2e74d68..68bf90b 100755
--- a/reinforcement_learning/multi_agent_training.py
+++ b/reinforcement_learning/multi_agent_training.py
@@ -181,7 +181,7 @@ def train_agent(train_params, train_env_params, eval_env_params, obs_params):
elif train_params.policy == "PPO":
policy = PPOPolicy(state_size, get_action_size(), use_replay_buffer=False, in_parameters=train_params)
elif train_params.policy == "DeadLockAvoidance":
- policy = DeadLockAvoidanceAgent(train_env, get_action_size())
+ policy = DeadLockAvoidanceAgent(train_env, get_action_size(), enable_eps=False)
elif train_params.policy == "DeadLockAvoidanceWithDecision":
# inter_policy = PPOPolicy(state_size, get_action_size(), use_replay_buffer=False, in_parameters=train_params)
inter_policy = DDDQNPolicy(state_size, get_action_size(), train_params)
--
GitLab