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))