lightning_template.utils.callbacks.set_wandb_logger_callback

Classes

SetWandbLoggerCallback

Abstract base class used to build new callbacks.

Module Contents

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

Abstract base class used to build new callbacks.

Subclass this class and override any of the relevant hooks

log_code_cfg = None

watch_model_cfg = None

log_code(trainer: lightning.pytorch.Trainer)

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

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