public
DenseNet
— struct
DenseNet(nblocks::NTuple{N, <:Integer};
growth_rate = 32, reduction = 0.5, nclasses = 1000)
Create a DenseNet model
(reference).
See also densenet
.
Arguments
nblocks
: number of dense blocks between transitionsgrowth_rate
: the output feature map growth rate of dense blocks (i.e.k
in the paper)reduction
: the factor by which the number of feature maps is scaled across each transitionnclasses
: the number of output classes
DenseNet(config::Integer = 121; pretrain = false, nclasses = 1000)
DenseNet(transition_config::NTuple{N,Integer})
Create a DenseNet model with specified configuration. Currently supported values are (121, 161, 169, 201)
(reference).
Set pretrain = true
to load the model with pre-trained weights for ImageNet.
Warning
DenseNet
does not currently support pretrained weights.
See also Metalhead.densenet
.