ConvMixer(mode::Symbol = :base; inchannels = 3, activation = gelu, nclasses = 1000)

Creates a ConvMixer model. (reference)

Arguments

• mode: the mode of the model, either :base, :small or :large
• inchannels: number of channels in the input
• activation: activation function used after the convolutional layers
• nclasses: number of classes in the output