ResMLP(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 ResMLP 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.