update

thu-ml · Mar 29, 2022 · 7f9e973 · 7f9e973
1 parent 03e932b
commit 7f9e973
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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.