cyto_dl.callbacks.grad_monitor module#

class cyto_dl.callbacks.grad_monitor.GradientLoggingCallback(grouping_level: int = 3)[source]#

Bases: Callback

on_train_epoch_end(trainer, pl_module)[source]#