From f77cddf37f914991d2095e593cc8296bb5ccaea0 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 5 Jan 2021 10:31:36 +0100
Subject: [PATCH] Policy updated
---
reinforcement_learning/multi_agent_training.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/reinforcement_learning/multi_agent_training.py b/reinforcement_learning/multi_agent_training.py
index 882cf86..6a34939 100755
--- a/reinforcement_learning/multi_agent_training.py
+++ b/reinforcement_learning/multi_agent_training.py
@@ -173,11 +173,11 @@ def train_agent(train_params, train_env_params, eval_env_params, obs_params):
policy = None
if False:
policy = DDDQNPolicy(state_size, get_action_size(), train_params)
- if False:
+ if True:
policy = PPOPolicy(state_size, get_action_size(), use_replay_buffer=False, in_parameters=train_params)
if False:
policy = DeadLockAvoidanceAgent(train_env, get_action_size())
- if True:
+ if False:
# 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)
policy = DeadLockAvoidanceWithDecisionAgent(train_env, state_size, get_action_size(), inter_policy)
--
GitLab