From bb160d28371e7b20b276cb550117dcaae9a1a2f1 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Fri, 17 Jun 2022 13:16:32 +0200
Subject: [PATCH] changed locations / names for logs

---
 run_tensorboard.sh       | 2 +-
 sb3_trl/trl_pg/trl_pg.py | 2 +-
 test.py                  | 1 +
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/run_tensorboard.sh b/run_tensorboard.sh
index 9623ac9..5197275 100755
--- a/run_tensorboard.sh
+++ b/run_tensorboard.sh
@@ -1,2 +1,2 @@
 #!/bin/bash
-tensorboard --logdir logs_tb/test
+tensorboard --logdir logs_tb
diff --git a/sb3_trl/trl_pg/trl_pg.py b/sb3_trl/trl_pg/trl_pg.py
index 9f524e6..1d813f5 100644
--- a/sb3_trl/trl_pg/trl_pg.py
+++ b/sb3_trl/trl_pg/trl_pg.py
@@ -322,7 +322,7 @@ class TRL_PG(OnPolicyAlgorithm):
         eval_env: Optional[GymEnv] = None,
         eval_freq: int = -1,
         n_eval_episodes: int = 5,
-        tb_log_name: str = "PPO",
+        tb_log_name: str = "TRL_PG",
         eval_log_path: Optional[str] = None,
         reset_num_timesteps: bool = True,
     ) -> "TRL_PG":
diff --git a/test.py b/test.py
index 9bfa910..dbe7a53 100644
--- a/test.py
+++ b/test.py
@@ -19,6 +19,7 @@ def main():
         "MlpPolicy",
         env,
         verbose=0,
+        tensorboard_log="./logs_tb/test/",
     )
     trl = TRL_PG(
         "MlpPolicy",