stable_ssl.trainers

SupervisedTrainer(data, module, hardware, ...)	Base class for training a supervised model.
JointEmbeddingTrainer(data, module, ...[, loss])	Base class for training a joint-embedding SSL model.
SelfDistillationTrainer(data, module, ...[, ...])	Base class for training a self-distillation SSL model.
DINOTrainer([warmup_temperature_teacher, ...])	DINO SSL model by [Caron et al., 2021].
JointEmbeddingPredictiveTrainer(data, ...[, ...])	Base class for training a joint-embedding predictive architecture.