private
convnext
— function
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 blockplanes
: list with configuration for number of output channels in each blockdrop_path_rate
: Stochastic depth rate.λ
: Initial value forLayerScale
(reference)nclasses
: number of output classes