Constructs a MobileNetV2 architecture from MobileNetV2: Inverted Residuals and Linear Bottlenecks. — model_mobilenet_v2 • torchvision

Constructs a MobileNetV2 architecture from MobileNetV2: Inverted Residuals and Linear Bottlenecks.

model_mobilenet_v2(pretrained = FALSE, progress = TRUE, ...)

Arguments

pretrained: (bool): If TRUE, returns a model pre-trained on ImageNet.
progress: (bool): If TRUE, displays a progress bar of the download to stderr.
...: Other parameters passed to the model implementation.

See also

Other models: model_alexnet(), model_inception_v3(), model_resnet, model_vgg