SelfDistillationTrainer

class stable_ssl.trainers.SelfDistillationTrainer(data, module, hardware, optim, logger, loss=None, **kwargs)

Bases: JointEmbeddingTrainer

Base class for training a self-distillation SSL model.

compute_loss()

Compute final loss as sum of SSL loss and classifier losses.