diff --git a/source/wheeledlab_rl/scripts/train_rl.py b/source/wheeledlab_rl/scripts/train_rl.py index d5674d3..85b7d39 100644 --- a/source/wheeledlab_rl/scripts/train_rl.py +++ b/source/wheeledlab_rl/scripts/train_rl.py @@ -48,6 +48,7 @@ def main(run_cfg: RunConfig): # TODO: Add SB3 config support import wandb run = wandb.init( project=log_cfg.wandb_project, + entity="thanandnow-university-of-washington", ) log_cfg.run_name = wandb.run.name @@ -121,4 +122,4 @@ def main(run_cfg: RunConfig): # TODO: Add SB3 config support if __name__ == "__main__": main() - simulation_app.close() \ No newline at end of file + simulation_app.close()