fedlab_core.models.senet

SENet in PyTorch.

SENet is the winner of ImageNet-2017. The paper is not released yet.

Module Contents

Classes

BasicBlock
PreActBlock
SENet

Functions

SENet18()
test()

class fedlab_core.models.senet.BasicBlock(in_planes, planes, stride=1)

Bases: torch.nn.Module

forward(self, x)

class fedlab_core.models.senet.PreActBlock(in_planes, planes, stride=1)

Bases: torch.nn.Module

forward(self, x)

class fedlab_core.models.senet.SENet(block, num_blocks, num_classes=10)

Bases: torch.nn.Module

_make_layer(self, block, planes, num_blocks, stride)

forward(self, x)

fedlab_core.models.senet.SENet18()

fedlab_core.models.senet.test()