`fedlab_core.models.densenet`¶

DenseNet in PyTorch.

Module Contents¶

Classes¶

`Bottleneck`
`Transition`
`DenseNet`

Functions¶

`DenseNet121`()
`DenseNet169`()
`DenseNet201`()
`DenseNet161`()
`densenet_cifar`()
`test_densenet`()

class fedlab_core.models.densenet.Bottleneck(in_planes, growth_rate)¶

Bases: torch.nn.Module

forward(self, x)¶

class fedlab_core.models.densenet.Transition(in_planes, out_planes)¶

Bases: torch.nn.Module

forward(self, x)¶

class fedlab_core.models.densenet.DenseNet(block, nblocks, growth_rate=12, reduction=0.5, num_classes=10)¶

Bases: torch.nn.Module

_make_dense_layers(self, block, in_planes, nblock)¶

forward(self, x)¶

fedlab_core.models.densenet.DenseNet121()¶

fedlab_core.models.densenet.DenseNet169()¶

fedlab_core.models.densenet.DenseNet201()¶

fedlab_core.models.densenet.DenseNet161()¶

fedlab_core.models.densenet.densenet_cifar()¶

fedlab_core.models.densenet.test_densenet()¶