Introduction

Tutorials

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: number of input channels.
  • drop_path_rate: Stochastic depth rate.
  • λ: Init value for LayerScale
  • nclasses: number of output classes

See also Metalhead.convnext.