Skip to content

Commit

Permalink
minor change
Browse files Browse the repository at this point in the history
  • Loading branch information
shivance committed Jan 22, 2023
1 parent 4fae8d6 commit 4735dff
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/convnets/unet.jl
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 4735dff

Please sign in to comment.