:mod:`fedlab_core.models.senet` =============================== .. py:module:: fedlab_core.models.senet .. autoapi-nested-parse:: SENet in PyTorch. SENet is the winner of ImageNet-2017. The paper is not released yet. Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: fedlab_core.models.senet.BasicBlock fedlab_core.models.senet.PreActBlock fedlab_core.models.senet.SENet Functions ~~~~~~~~~ .. autoapisummary:: fedlab_core.models.senet.SENet18 fedlab_core.models.senet.test .. class:: BasicBlock(in_planes, planes, stride=1) Bases: :class:`torch.nn.Module` .. method:: forward(self, x) .. class:: PreActBlock(in_planes, planes, stride=1) Bases: :class:`torch.nn.Module` .. method:: forward(self, x) .. class:: SENet(block, num_blocks, num_classes=10) Bases: :class:`torch.nn.Module` .. method:: _make_layer(self, block, planes, num_blocks, stride) .. method:: forward(self, x) .. function:: SENet18() .. function:: test()