Introduction

Tutorials

Developer guide

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.
  • λ: Initial value for LayerScale (reference)
  • nclasses: number of output classes