diff --git a/src/convnets/unet.jl b/src/convnets/unet.jl index 333577b07..458a37f12 100644 --- a/src/convnets/unet.jl +++ b/src/convnets/unet.jl @@ -87,8 +87,8 @@ end UNet(imsize::Dims{2} = (256, 256), inchannels::Integer = 3, outplanes::Integer = 3, encoder_backbone = Metalhead.backbone(DenseNet(121)); pretrain::Bool = false) -Creates a UNet model with an encoder built of specified backbone. -By default it uses [`DenseNet`](@ref) backbone, but any ResNet-like Metalhead model can be used for the encoder +Creates a UNet model with an encoder built of specified backbone. By default it uses +[`DenseNet`](@ref) backbone, but any ResNet-like Metalhead model can be used for the encoder. ([reference](https://arxiv.org/abs/1505.04597)). # Arguments