solo-learn
stable
Getting started
Installation
Methods available
Extra flavor
Tutorials
An Overview
Offline Linear Eval
Adding New Methods
Adding New Methods (Momentum Version)
Args
Args
Utils
Utils
Methods
BaseMethod
BaseMomentumMethod
LinearModel
add_model_specific_args
configure_optimizers
forward
shared_step
training_step
validation_step
on_validation_epoch_end
Barlow Twins
BYOL
DeepClusterV2
DINO
MAE
MoCo-V2
MoCo-V3
NNBYOL
NNCLR
NNSiam
ReSSL
SimCLR
SimSiam
SwAV
VIbCReg
VICReg
W-MSE
Losses
Barlow Twins
BYOL
DeepClusterV2
DINO
MAE
MoCo-V2
MoCo-V3
NNCLR
ReSSL
SimCLR
SimSiam
SwAV
VIbCReg
VICReg
W-MSE
Data
Classification dataloaders
Augmentations
Transformation Pipelines
Pretrain dataloader
DALI dataloaders
solo-learn
LinearModel
Edit on GitHub
LinearModel
add_model_specific_args
configure_optimizers
forward
shared_step
training_step
validation_step
on_validation_epoch_end