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