`fedlab_core.models.mobilenet`¶

MobileNet in PyTorch.

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

Module Contents¶

Classes¶

`Block`	Depthwise conv + Pointwise conv
`MobileNet`

Functions¶

class fedlab_core.models.mobilenet.Block(in_planes, out_planes, stride=1)¶

Bases: torch.nn.Module

Depthwise conv + Pointwise conv

forward(self, x)¶

class fedlab_core.models.mobilenet.MobileNet(num_classes=10)¶

Bases: torch.nn.Module

cfg = [64, [128, 2], 128, [256, 2], 256, [512, 2], 512, 512, 512, 512, 512, [1024, 2], 1024]¶

_make_layers(self, in_planes)¶

forward(self, x)¶

fedlab_core.models.mobilenet.test()¶