From 043e832db5bd2c5fabc33e6c977a62ba310eda68 Mon Sep 17 00:00:00 2001 From: Adrian Egli <adrian.egli@sbb.ch> Date: Sat, 7 Nov 2020 00:01:15 +0100 Subject: [PATCH] . --- run.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/run.py b/run.py index aaa4bab..d610fab 100644 --- a/run.py +++ b/run.py @@ -137,9 +137,9 @@ while True: action = agent_last_action[agent] nb_hit += 1 else: - action = policy.act(observation[agent], eps=0.0) - #if observation[agent][26] == 1: - # action = RailEnvActions.STOP_MOVING + action = policy.act(observation[agent], eps=0.01) + if observation[agent][26] == 1: + action = RailEnvActions.STOP_MOVING action_dict[agent] = action -- GitLab