solo-learn
latest
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
All4One
Barlow Twins
BYOL
DeepClusterV2
DINO
MAE
MoCo-V2
MoCo-V3
NNBYOL
NNCLR
NNSiam
ReSSL
SimCLR
SimSiam
SwAV
add_model_specific_args
learnable_params
on_train_start
forward
get_assignments
training_step
on_after_backward
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
SwAV
Edit on GitHub
SwAV
add_model_specific_args
learnable_params
on_train_start
forward
get_assignments
training_step
on_after_backward
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds