ModuleSummary

ModuleSummary#

class stable_ssl.callbacks.ModuleSummary[source]#

Bases: Callback

Callback for logging module summaries in a formatted table.

setup(trainer, pl_module, stage)[source]#

Called when fit, validate, test, predict, or tune begins.