Introduction

Tutorials

Developer guide

API Reference

public ConvNeXtstruct

ConvNeXt(mode::Symbol = :base; inchannels = 3, drop_path_rate = 0., λ = 1f-6, nclasses = 1000)

Creates a ConvNeXt model. (reference)

Arguments:

  • inchannels: The number of channels in the input. The default value is 3.
  • drop_path_rate: Stochastic depth rate.
  • λ: Init value for LayerScale
  • nclasses: number of output classes

See also Metalhead.convnext.