stable_ssl.modules

load_backbone(name, num_classes[, weights, ...])

Load a backbone model.

MLP(sizes[, activation, batch_norm])	Multi-layer perceptron.
Resnet9([in_channels, num_classes])	ResNet-9 model.
ConvMixer([in_channels, num_classes, dim, ...])	ConvMixer model from [Trockman and Kolter, 2022].