private convmixer — function
convmixer(planes, depth; inchannels = 3, kernel_size = (9, 9), patch_size::Dims{2} = 7,
activation = gelu, nclasses = 1000)
Creates a ConvMixer model. (reference)
Arguments
planes: number of planes in the output of each blockdepth: number of layersinchannels: The number of channels in the input. The default value is 3.kernel_size: kernel size of the convolutional layerspatch_size: size of the patchesactivation: activation function used after the convolutional layersnclasses: number of classes in the output