model_mobilenet_v2: Constructs a MobileNetV2 architecture from MobileNetV2:...

In mlverse/torchvision: Models, Datasets and Transformations for Images

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

Description

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

Usage

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

mlverse/torchvision documentation built on Sept. 18, 2024, 4:03 p.m.