norse.torch.models.mobilenet

Functions

mobilenet_v2([pretrained, progress])

Constructs a MobileNetV2 architecture from "MobileNetV2: Inverted Residuals and Linear Bottlenecks". Args: pretrained (bool): If True, returns a model pre-trained on ImageNet progress (bool): If True, displays a progress bar of the download to stderr.

Classes

ConvBNReLU(in_planes, out_planes, norm_layer)
InvertedResidual(inp, oup, stride, ...)
MobileNetV2([num_classes, width_mult, ...])