OnlineProbe

class stable_ssl.callbacks.OnlineProbe(name: str, pl_module: LightningModule, input: str, target: str, probe: Module, loss_fn: callable, optimizer: Optimizer | None = None, scheduler: LRScheduler | None = None, accumulate_grad_batches: int = 1, metrics: dict | tuple | list | Metric | None = None, early_stopping: EarlyStopping | None = None)

Bases: Callback

Attaches a MLP for fine-tuning using the standard self-supervised protocol.

on_validation_end(trainer, pl_module)

Called when the validation loop ends.