Merge pull request #26 from vwxyzjn/log-algo

Logs the algo name in wandb
ikostrikov · May 23, 2022 · 6550d6f · 6550d6f
2 parents 6a37d80 + 39fb70e
commit 6550d6f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/examples/train.py b/examples/train.py
@@ -56,6 +56,7 @@ def main(_):
             monitor_gym=True,
             save_code=True,
         )
+        wandb.config.update({"algo": algo})
 
     summary_writer = SummaryWriter(
         os.path.join(FLAGS.save_dir, run_name))

diff --git a/pyproject.toml b/pyproject.toml
@@ -37,6 +37,7 @@ imageio-ffmpeg = "^0.4.7"
 mujoco-py = "^2.1.2"
 d4rl = {git = "https://github.com/ikostrikov/d4rl.git"}
 dm-control = "^1.0.2"
+wandb = "^0.12.16"
 
 [tool.poetry.dev-dependencies]