From 7f9e9734996069b7e2b332ac04aa5bf891dc9b3b Mon Sep 17 00:00:00 2001 From: Jiayi Weng Date: Tue, 29 Mar 2022 19:34:03 -0400 Subject: [PATCH] update --- tianshou/utils/logger/tensorboard.py | 6 +++--- tianshou/utils/logger/wandb.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) 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.