mridc.collections.common.callbacks package

Submodules

mridc.collections.common.callbacks.callbacks module

class mridc.collections.common.callbacks.callbacks.LogEpochTimeCallback

Bases: Callback

Simple callback that logs how long each epoch takes, in seconds, to a pytorch lightning log

on_train_epoch_end(trainer, pl_module)

Called at the end of each epoch.

on_train_epoch_start(trainer, pl_module)

Called at the start of each epoch.

Module contents