Toggle Light / Dark / Auto color theme

Toggle table of contents sidebar

anomalib.utils.callbacks.graph

Log model graph to respective logger.

Module Contents

Classes

GraphLogger

Log model graph to respective logger.

class anomalib.utils.callbacks.graph.GraphLogger

Bases: pytorch_lightning.Callback

Log model graph to respective logger.

on_train_start(self, trainer: pytorch_lightning.Trainer, pl_module: pytorch_lightning.LightningModule) → None

Log model graph to respective logger.

Parameters

• trainer – Trainer object which contans reference to loggers.

• pl_module – LightningModule object which is logged.

on_train_end(self, trainer: pytorch_lightning.Trainer, pl_module: pytorch_lightning.LightningModule) → None

Unwatch model if configured for wandb and log it model graph in Tensorboard if specified.

Parameters

• trainer – Trainer object which contans reference to loggers.

• pl_module – LightningModule object which is logged.