From 90d7b94abb6628835ccaab2632aef7d68e294416 Mon Sep 17 00:00:00 2001
From: Guillaume Mollard <guillaume.mollard2@gmail.com>
Date: Tue, 14 May 2019 13:40:14 +0200
Subject: [PATCH] small fix
---
train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.py b/train.py
index 59db8d0..4e887c8 100644
--- a/train.py
+++ b/train.py
@@ -86,7 +86,7 @@ def train(config):
"policies_to_train": list(policy_graphs.keys())}
agent_config["horizon"] = 50
- ppo_trainer = PPOAgent(env=f"railenv_", config=agent_config)
+ ppo_trainer = PPOAgent(env=f"railenv", config=agent_config)
for i in range(100000 + 2):
print("== Iteration", i, "==")
--
GitLab