lightning_template.utils.callbacks.set_wandb_logger_callback

Module Contents

Classes

SetWandbLoggerCallback

Set wandb logger when training starts.

class lightning_template.utils.callbacks.set_wandb_logger_callback.SetWandbLoggerCallback(log_code_cfg: dict | bool | None = None, watch_model_cfg: dict | bool | None = None)

Bases: lightning.pytorch.Callback

Set wandb logger when training starts.

setup(trainer: lightning.pytorch.Trainer, pl_module: lightning.pytorch.LightningModule, stage: str | None = None) → None

Called when fit, validate, test, predict, or tune begins.