Logs the algo name in wandb

soroushmehr · May 22, 2022 · fbcde9d · fbcde9d
1 parent 6a37d80
commit fbcde9d
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))