torchvision.models.MobileNetV2#
- class torchvision.models.MobileNetV2(num_classes: int = 1000, width_mult: float = 1.0, inverted_residual_setting: Optional[List[List[int]]] = None, round_nearest: int = 8, block: Optional[Callable[[...], torch.nn.modules.module.Module]] = None, norm_layer: Optional[Callable[[...], torch.nn.modules.module.Module]] = None, dropout: float = 0.2)[源代码]#