:py:mod:`lightning_template.utils.callbacks.set_wandb_logger_callback`
======================================================================

.. py:module:: lightning_template.utils.callbacks.set_wandb_logger_callback


Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   lightning_template.utils.callbacks.set_wandb_logger_callback.SetWandbLoggerCallback




.. py:class:: SetWandbLoggerCallback(log_code_cfg: Optional[Union[dict, bool]] = None, watch_model_cfg: Optional[Union[dict, bool]] = None)


   Bases: :py:obj:`lightning.pytorch.Callback`

   Set wandb logger when training starts.

   .. py:method:: setup(trainer: lightning.pytorch.Trainer, pl_module: lightning.pytorch.LightningModule, stage: Optional[str] = None) -> None

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