private
convnextblock
— function
convnextblock(planes, drop_path_rate = 0., λ = 1f-6)
Creates a single block of ConvNeXt. (reference)
Arguments:
planes
: number of input channels.drop_path_rate
: Stochastic depth rate.λ
: Initial value forLayerScale