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
add_model_specific_args
learnable_params
configure_optimizers
forward
multicrop_forward
_base_shared_step
training_step
validation_step
on_validation_epoch_end
BaseMomentumMethod
learnable_params
momentum_pairs
add_model_specific_args
on_train_start
momentum_forward
_shared_step_momentum
training_step
on_train_batch_end
validation_step
on_validation_epoch_end
LinearModel
All4One
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
BaseMethod
Edit on GitHub
BaseMethod
add_model_specific_args
learnable_params
configure_optimizers
forward
multicrop_forward
_base_shared_step
training_step
validation_step
on_validation_epoch_end
BaseMomentumMethod
learnable_params
momentum_pairs
add_model_specific_args
on_train_start
momentum_forward
_shared_step_momentum
training_step
on_train_batch_end
validation_step
on_validation_epoch_end
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds