private mixerblock — function
mixerblock(planes, npatches; mlp_ratio = (0.5, 4.0), mlp_layer = mlp_block,
dropout = 0., drop_path_rate = 0., activation = gelu)
Creates a feedforward block for the MLPMixer architecture. (reference)
Arguments:
planes: the number of planes in the blocknpatches: the number of patches of the inputmlp_ratio: number(s) that determine(s) the number of hidden channels in the token mixing MLP and/or the channel mixing MLP as a ratio to the number of planes in the block.mlp_layer: the MLP layer to use in the blockdropout: the dropout rate to use in the MLP blocksdrop_path_rate: Stochastic depth rateactivation: the activation function to use in the MLP blocks