Logs the algo name in wandb

ikostrikov · May 13, 2022 · e6ec9d9 · e6ec9d9
1 parent 52b3653
commit e6ec9d9
Showing 1 changed file with 1 addition 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))