Skip to main content

epoch_callbacks

Pytorch Lightning Callbacks to trigger Bitfount Epoch Hooks.

Classes​

EpochCallbacks​

class EpochCallbacks():

Pytorch Lightning Callbacks to trigger Bitfount Epoch Hooks.

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.