Skip to content

Commit

Permalink
Update composer/loggers/wandb_logger.py
Browse files Browse the repository at this point in the history
Co-authored-by: Mihir Patel <mihir.v.patel7@gmail.com>
  • Loading branch information
snarayan21 and mvpatel2000 authored Dec 3, 2024
1 parent fad0230 commit f685a04
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion composer/loggers/wandb_logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ class WandBLogger(LoggerDestination):
rank_zero_only (bool, optional): Whether to log only on the rank-zero process.
When logging `artifacts <https://docs.wandb.ai/ref/python/artifact>`_, it is
highly recommended to log on all ranks. Artifacts from ranks ≥1 will not be
stored, which may discard pertinent information.
stored, which may discard pertinent information (default: ``True``).
init_kwargs (dict[str, Any], optional): Any additional init kwargs
``wandb.init`` (see
`WandB documentation <https://docs.wandb.ai/ref/python/init>`_).
Expand Down

0 comments on commit f685a04

Please sign in to comment.