composer.loggers.wandb_logger#

Log to Weights and Biases (https://wandb.ai/)

Classes

WandBLogger

Log to Weights and Biases (https://wandb.ai/)
class composer.loggers.wandb_logger.WandBLogger(log_artifacts=False, rank_zero_only=True, init_params=None)[source]#

Bases: composer.loggers.logger_destination.LoggerDestination

Log to Weights and Biases (https://wandb.ai/)

Parameters

  • log_artifacts (bool, optional) โ€“ Whether to log artifacts (Default: False).

  • 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. For example, when using Deepspeed ZeRO, it would be impossible to restore from checkpoints without artifacts from all ranks (default: False).

  • init_params (Dict[str, Any], optional) โ€“ Parameters to pass into wandb.init (see WandB documentation).