WandBLoggerBackendHparams

class composer.loggers.logger_hparams.WandBLoggerBackendHparams(project: Optional[str] = None, name: Optional[str] = None, entity: Optional[str] = None, tags: str = '', log_artifacts: bool = False, log_artifacts_every_n_batches: int = 100, extra_init_params: Dict[str, Union[str, float, int, None, List[JSON], Dict[str, JSON]]] = <factory>)[source]

Bases: composer.loggers.logger_hparams.BaseLoggerBackendHparams

WandBLoggerBackend hyperparameters.

Parameters
  • project (str, optional) – Weights and Biases project name.

  • name (str, optional) – Weights and Biases run name.

  • entity (str, optional) – Weights and Biases entity name.

  • tags (str, optional) – Comma-seperated list of tags to add to the run.

  • log_artifacts (bool, optional) – Whether to log artifacts. Defaults to False.

  • log_artifacts_every_n_batches (int, optional) – Only applicable if log_artifacts is True.

  • extra_init_params (JSON Dictionary, optional) – Extra parameters to pass into wandb.init().

initialize_object(config: Optional[Dict[str, Any]] = None) WandBLoggerBackend[source]

Initializes the logger.

The config is flattened and stored as wandb.run.config. The list of algorithms in the config are appended to wandb.run.tags.

Parameters

config (Optional[Dict[str, Any]], optional) – The configuration used by the trainer.

Returns

WandBLoggerBackend – An instance of WandBLoggerBackend.