TrainerInfo

TrainerInfo#

class stable_ssl.callbacks.TrainerInfo[source]#

Bases: Callback

Callback for linking trainer to DataModule and providing extra information.

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

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