:mod:`fedlab_core.models.lenet`
===============================

.. py:module:: fedlab_core.models.lenet

.. autoapi-nested-parse::

   LeNet in PyTorch.



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

Classes
~~~~~~~

.. autoapisummary::

   fedlab_core.models.lenet.Lenet
   fedlab_core.models.lenet.LeNet



.. class:: Lenet


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

   .. method:: forward(self, x)



.. class:: LeNet(out_dim=10, in_channel=1, img_sz=28)


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

   .. method:: features(self, x)


   .. method:: logits(self, x)


   .. method:: forward(self, x)