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: pytorch_lightning.trainer.trainer.Trainer,    pl_module: pytorch_lightning.core.module.LightningModule,) ‑> None:

Called when the train epoch ends.

on_train_epoch_start

def on_train_epoch_start(    self,    trainer: pytorch_lightning.trainer.trainer.Trainer,    pl_module: pytorch_lightning.core.module.LightningModule,) ‑> None:

Called when the train epoch begins.

---