From f685a04ee3c44e86e9e8d5a537442f168413323d Mon Sep 17 00:00:00 2001 From: Saaketh Narayan Date: Tue, 3 Dec 2024 13:15:34 -0500 Subject: [PATCH] Update composer/loggers/wandb_logger.py Co-authored-by: Mihir Patel --- composer/loggers/wandb_logger.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/composer/loggers/wandb_logger.py b/composer/loggers/wandb_logger.py index 3fb0a13bee..6412114168 100644 --- a/composer/loggers/wandb_logger.py +++ b/composer/loggers/wandb_logger.py @@ -44,7 +44,7 @@ class WandBLogger(LoggerDestination): rank_zero_only (bool, optional): Whether to log only on the rank-zero process. When logging `artifacts `_, 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 `_).