@@ -7,7 +7,7 @@ Creates a single block of ConvNeXt.
77# Arguments
88
99 - `planes`: number of input channels.
10- - `drop_path_rate `: Stochastic depth rate .
10+ - `stochastic_depth_prob `: Stochastic depth probability .
1111 - `layerscale_init`: Initial value for [`Metalhead.LayerScale`](@ref)
1212"""
1313function convnextblock (planes:: Integer , stochastic_depth_prob = 0.0 ,
@@ -33,13 +33,8 @@ Creates the layers for a ConvNeXt model.
3333
3434 - `depths`: list with configuration for depth of each block
3535 - `planes`: list with configuration for number of output channels in each block
36- <<<<<<< HEAD
3736 - `stochastic_depth_prob`: Stochastic depth probability.
3837 - `layerscale_init`: Initial value for [`LayerScale`](@ref)
39- =======
40- - `drop_path_rate`: Stochastic depth rate.
41- - `layerscale_init`: Initial value for [`Metalhead.LayerScale`](@ref)
42- >>>>>>> 5ac81ce (Use `@autodocs` and fix references)
4338 ([reference](https://arxiv.org/abs/2103.17239))
4439 - `inchannels`: number of input channels.
4540 - `nclasses`: number of output classes
@@ -101,13 +96,10 @@ Creates a ConvNeXt model.
10196 - `inchannels`: number of input channels
10297 - `nclasses`: number of output classes
10398
104- <<<<<<< HEAD
10599!!! warning
106100
107101 `ConvNeXt` does not currently support pretrained weights.
108102
109- =======
110- >>>>>>> 5ac81ce (Use `@autodocs` and fix references)
111103See also [`Metalhead.convnext`](@ref).
112104"""
113105struct ConvNeXt
0 commit comments