gMLP(size::Symbol = :base; patch_size::Dims{2} = (16, 16),
     imsize::Dims{2} = (224, 224), drop_path_rate = 0., nclasses = 1000)

Creates a model with the gMLP architecture. (reference).

Arguments

• size: the size of the model - one of small, base, large or huge
• patch_size: the size of the patches
• imsize: the size of the input image
• drop_path_rate: Stochastic depth rate
• nclasses: number of output classes

See also Metalhead.mlpmixer.