diff --git a/tianshou/utils/logger/wandb.py b/tianshou/utils/logger/wandb.py index c8e78d5..8e8e05d 100644 --- a/tianshou/utils/logger/wandb.py +++ b/tianshou/utils/logger/wandb.py @@ -54,6 +54,7 @@ class WandbLogger(BaseLogger): entity: Optional[str] = None, run_id: Optional[str] = None, config: Optional[argparse.Namespace] = None, + monitor_gym: bool = True, ) -> None: super().__init__(train_interval, test_interval, update_interval) self.last_save_step = -1 @@ -70,7 +71,7 @@ class WandbLogger(BaseLogger): resume="allow", entity=entity, sync_tensorboard=True, - monitor_gym=True, + monitor_gym=monitor_gym, config=config, # type: ignore ) if not wandb.run else wandb.run self.wandb_run._label(repo="tianshou") # type: ignore