lightning_template.utils.callbacks.set_wandb_logger_callback
============================================================

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


Classes
-------

.. autoapisummary::

   lightning_template.utils.callbacks.set_wandb_logger_callback.SetWandbLoggerCallback


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

.. 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`


   Abstract base class used to build new callbacks.

   Subclass this class and override any of the relevant hooks



   .. py:attribute:: log_code_cfg
      :value: None



   .. py:attribute:: watch_model_cfg
      :value: None



   .. py:method:: log_code(trainer: lightning.pytorch.Trainer)


   .. 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.