epoch_callbacks

Pytorch Lightning Callbacks to trigger Bitfount Epoch Hooks.

Classes

EpochCallbacks

class EpochCallbacks():

Pytorch Lightning Callbacks to trigger Bitfount Epoch Hooks.

Ancestors

• pytorch_lightning.callbacks.callback.Callback

Methods

---

on_train_epoch_end

def on_train_epoch_end(self, trainer: pl.Trainer, pl_module: pl.LightningModule) ‑> None:

Called when the train epoch ends.

on_train_epoch_start

def on_train_epoch_start(    self, trainer: pl.Trainer, pl_module: pl.LightningModule,) ‑> None:

Called when the train epoch begins.

---