diff --git a/tianshou/utils/logger/tensorboard.py b/tianshou/utils/logger/tensorboard.py index c72dfb3e5..843ff012e 100644 --- a/tianshou/utils/logger/tensorboard.py +++ b/tianshou/utils/logger/tensorboard.py @@ -17,8 +17,8 @@ class TensorboardLogger(BaseLogger): :param int update_interval: the log interval in log_update_data(). Default to 1000. :param int save_interval: the save interval in save_data(). Default to 1 (save at the end of each epoch). - :param bool write_flush: whether to flush tensorboard result after each "add_*" - operation. Default to True. + :param bool write_flush: whether to flush tensorboard result after each + add_scalar operation. Default to True. """ def __init__( @@ -28,7 +28,7 @@ def __init__( test_interval: int = 1, update_interval: int = 1000, save_interval: int = 1, - write_flush: bool = True + write_flush: bool = True, ) -> None: super().__init__(train_interval, test_interval, update_interval) self.save_interval = save_interval diff --git a/tianshou/utils/logger/wandb.py b/tianshou/utils/logger/wandb.py index 062a26a6c..e63a7bc7f 100644 --- a/tianshou/utils/logger/wandb.py +++ b/tianshou/utils/logger/wandb.py @@ -33,8 +33,8 @@ class WandbLogger(BaseLogger): Default to 1000. :param int save_interval: the save interval in save_data(). Default to 1 (save at the end of each epoch). - :param bool write_flush: whether to flush tensorboard result after each "add_*" - operation. Default to True. + :param bool write_flush: whether to flush tensorboard result after each + add_scalar operation. Default to True. :param str project: W&B project name. Default to "tianshou". :param str name: W&B run name. Default to None. If None, random name is assigned. :param str entity: W&B team/organization name. Default to None.