From e6ec9d9f9409152edb6553812aa9752b28e4b287 Mon Sep 17 00:00:00 2001 From: Costa Huang Date: Thu, 12 May 2022 22:57:01 -0400 Subject: [PATCH] Logs the algo name in wandb --- examples/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/train.py b/examples/train.py index eb6c0e1..cf40d11 100644 --- 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))