Introduction

Tutorials

Developer guide

API Reference

private convmixerfunction

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 block
  • depth: number of layers
  • inchannels: The number of channels in the input. The default value is 3.
  • kernel_size: kernel size of the convolutional layers
  • patch_size: size of the patches
  • activation: activation function used after the convolutional layers
  • nclasses: number of classes in the output