:mod:`fedlab_core.models.densenet`
==================================

.. py:module:: fedlab_core.models.densenet

.. autoapi-nested-parse::

   DenseNet in PyTorch.



Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   fedlab_core.models.densenet.Bottleneck
   fedlab_core.models.densenet.Transition
   fedlab_core.models.densenet.DenseNet



Functions
~~~~~~~~~

.. autoapisummary::

   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


.. class:: Bottleneck(in_planes, growth_rate)


   Bases: :class:`torch.nn.Module`

   .. method:: forward(self, x)



.. class:: Transition(in_planes, out_planes)


   Bases: :class:`torch.nn.Module`

   .. method:: forward(self, x)



.. class:: DenseNet(block, nblocks, growth_rate=12, reduction=0.5, num_classes=10)


   Bases: :class:`torch.nn.Module`

   .. method:: _make_dense_layers(self, block, in_planes, nblock)


   .. method:: forward(self, x)



.. function:: DenseNet121()


.. function:: DenseNet169()


.. function:: DenseNet201()


.. function:: DenseNet161()


.. function:: densenet_cifar()


.. function:: test_densenet()