:mod:`fedlab_core.models.mobilenetv2`
=====================================

.. py:module:: fedlab_core.models.mobilenetv2

.. autoapi-nested-parse::

   MobileNetV2 in PyTorch.

   See the paper "Inverted Residuals and Linear Bottlenecks:
   Mobile Networks for Classification, Detection and Segmentation" for more details.



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

Classes
~~~~~~~

.. autoapisummary::

   fedlab_core.models.mobilenetv2.Block
   fedlab_core.models.mobilenetv2.MobileNetV2



Functions
~~~~~~~~~

.. autoapisummary::

   fedlab_core.models.mobilenetv2.test


.. class:: Block(in_planes, out_planes, expansion, stride)


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

   expand + depthwise + pointwise

   .. method:: forward(self, x)



.. class:: MobileNetV2(num_classes=10)


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

   .. attribute:: cfg
      :annotation: = [[1, 16, 1, 1], [6, 24, 2, 1], [6, 32, 3, 2], [6, 64, 4, 2], [6, 96, 3, 1], [6, 160, 3, 2], [6, 320, 1, 1]]

      

   .. method:: _make_layers(self, in_planes)


   .. method:: forward(self, x)



.. function:: test()