Introduction

Tutorials

API Reference

private convnextfunction

convnext(depths, planes; inchannels = 3, drop_path_rate = 0., λ = 1f-6, nclasses = 1000)

Creates the layers for a ConvNeXt model. (reference)

Arguments:

  • inchannels: number of input channels.
  • depths: list with configuration for depth of each block
  • planes: list with configuration for number of output channels in each block
  • drop_path_rate: Stochastic depth rate.
  • λ: Init value for LayerScale
  • nclasses: number of output classes