ConvMixer

ConvMixer#

class stable_ssl.backbone.ConvMixer(in_channels=3, num_classes=10, dim=64, depth=6, kernel_size=9, patch_size=7)[source]#

Bases: Module

ConvMixer model from [Trockman and Kolter, 2022].

forward(xb)[source]#

Forward pass.