diff --git a/torchtnt/utils/loggers/tensorboard.py b/torchtnt/utils/loggers/tensorboard.py index 7f6b00fb7f..e748097152 100644 --- a/torchtnt/utils/loggers/tensorboard.py +++ b/torchtnt/utils/loggers/tensorboard.py @@ -93,7 +93,7 @@ def log_dict(self, payload: Mapping[str, Scalar], step: int) -> None: Args: payload (dict): dictionary of tag name and scalar value - step (int, Optional): step value to record + step (int): step value to record """ if self._writer: @@ -106,7 +106,7 @@ def log(self, name: str, data: Scalar, step: int) -> None: Args: name (string): tag name used to group scalars data (float/int/Tensor): scalar data to log - step (int, optional): step value to record + step (int): step value to record """ if self._writer: