From d53cbf44e8e5facfbeb82d2af1fdeda7dece28ec Mon Sep 17 00:00:00 2001 From: Saransh Date: Mon, 5 Sep 2022 20:07:24 +0530 Subject: [PATCH] Rebase errors --- src/convnets/convnext.jl | 8 -------- src/convnets/resnets/core.jl | 8 -------- src/convnets/resnets/resnet.jl | 4 ---- src/convnets/resnets/seresnet.jl | 4 ---- 4 files changed, 24 deletions(-) diff --git a/src/convnets/convnext.jl b/src/convnets/convnext.jl index 0f17b39cb..dd21e3268 100644 --- a/src/convnets/convnext.jl +++ b/src/convnets/convnext.jl @@ -33,13 +33,8 @@ Creates the layers for a ConvNeXt model. - `depths`: list with configuration for depth of each block - `planes`: list with configuration for number of output channels in each block -<<<<<<< HEAD - `stochastic_depth_prob`: Stochastic depth probability. - `layerscale_init`: Initial value for [`LayerScale`](@ref) -======= - - `drop_path_rate`: Stochastic depth rate. - - `layerscale_init`: Initial value for [`Metalhead.LayerScale`](@ref) ->>>>>>> 5ac81ce (Use `@autodocs` and fix references) ([reference](https://arxiv.org/abs/2103.17239)) - `inchannels`: number of input channels. - `nclasses`: number of output classes @@ -101,13 +96,10 @@ Creates a ConvNeXt model. - `inchannels`: number of input channels - `nclasses`: number of output classes -<<<<<<< HEAD !!! warning `ConvNeXt` does not currently support pretrained weights. -======= ->>>>>>> 5ac81ce (Use `@autodocs` and fix references) See also [`Metalhead.convnext`](@ref). """ struct ConvNeXt diff --git a/src/convnets/resnets/core.jl b/src/convnets/resnets/core.jl index ba8899e12..c5b2f1c7d 100644 --- a/src/convnets/resnets/core.jl +++ b/src/convnets/resnets/core.jl @@ -66,11 +66,7 @@ used to build the block for the model, see [`Metalhead.bottleneck_builder`](@ref - `revnorm`: set to `true` to place the normalisation layer before the convolution - `drop_block`: the drop block layer - `drop_path`: the drop path layer -<<<<<<< HEAD - `attn_fn`: the attention function to use. See [`squeeze_excite`](@ref) for an example. -======= - - `attn_fn`: the attention function to use. See [`Metalhead.squeeze_excite`](@ref) for an example. ->>>>>>> 5ac81ce (Use `@autodocs` and fix references) """ function bottleneck(inplanes::Integer, planes::Integer; stride::Integer, cardinality::Integer = 1, base_width::Integer = 64, @@ -235,11 +231,7 @@ const RESNET_SHORTCUTS = Dict(:A => (downsample_identity, downsample_identity), resnet_stem(; stem_type = :default, inchannels::Integer = 3, replace_stem_pool = false, norm_layer = BatchNorm, activation = relu) -<<<<<<< HEAD Builds a stem to be used in a ResNet model. See the `stem` argument of [`resnet`](@ref) for details -======= -Builds a stem to be used in a ResNet model. See the `stem` argument of [`Metalhead.resnet`](@ref) for details ->>>>>>> 5ac81ce (Use `@autodocs` and fix references) on how to use this function. # Arguments diff --git a/src/convnets/resnets/resnet.jl b/src/convnets/resnets/resnet.jl index 62a441d73..a8c12eda8 100644 --- a/src/convnets/resnets/resnet.jl +++ b/src/convnets/resnets/resnet.jl @@ -48,11 +48,7 @@ The number of channels in outer 1x1 convolutions is the same. - `inchannels`: The number of input channels. - `nclasses`: The number of output classes -<<<<<<< HEAD Advanced users who want more configuration options will be better served by using [`resnet`](@ref). -======= -Advanced users who want more configuration options will be better served by using [`Metalhead.resnet`](@ref). ->>>>>>> 5ac81ce (Use `@autodocs` and fix references) """ struct WideResNet layers::Any diff --git a/src/convnets/resnets/seresnet.jl b/src/convnets/resnets/seresnet.jl index 83d6cb711..44e32083d 100644 --- a/src/convnets/resnets/seresnet.jl +++ b/src/convnets/resnets/seresnet.jl @@ -58,11 +58,7 @@ Creates a SEResNeXt model with the specified depth, cardinality, and base width. `SEResNeXt` does not currently support pretrained weights. -<<<<<<< HEAD Advanced users who want more configuration options will be better served by using [`resnet`](@ref). -======= -Advanced users who want more configuration options will be better served by using [`Metalhead.resnet`](@ref). ->>>>>>> 5ac81ce (Use `@autodocs` and fix references) """ struct SEResNeXt layers::Any