:mod:`fedlab_core.models.mobilenet`
===================================

.. py:module:: fedlab_core.models.mobilenet

.. autoapi-nested-parse::

   MobileNet in PyTorch.

   See the paper "MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications"
   for more details.



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

Classes
~~~~~~~

.. autoapisummary::

   fedlab_core.models.mobilenet.Block
   fedlab_core.models.mobilenet.MobileNet



Functions
~~~~~~~~~

.. autoapisummary::

   fedlab_core.models.mobilenet.test


.. class:: Block(in_planes, out_planes, stride=1)


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

   Depthwise conv + Pointwise conv

   .. method:: forward(self, x)



.. class:: MobileNet(num_classes=10)


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

   .. attribute:: cfg
      :annotation: = [64, [128, 2], 128, [256, 2], 256, [512, 2], 512, 512, 512, 512, 512, [1024, 2], 1024]

      

   .. method:: _make_layers(self, in_planes)


   .. method:: forward(self, x)



.. function:: test()