mobilenetv2.rst 697 B

123456789101112131415161718192021222324
  1. MobileNet V2
  2. ============
  3. .. currentmodule:: torchvision.models
  4. The MobileNet V2 model is based on the `MobileNetV2: Inverted Residuals and Linear
  5. Bottlenecks <https://arxiv.org/abs/1801.04381>`__ paper.
  6. Model builders
  7. --------------
  8. The following model builders can be used to instantiate a MobileNetV2 model, with or
  9. without pre-trained weights. All the model builders internally rely on the
  10. ``torchvision.models.mobilenetv2.MobileNetV2`` base class. Please refer to the `source
  11. code
  12. <https://github.com/pytorch/vision/blob/main/torchvision/models/mobilenetv2.py>`_ for
  13. more details about this class.
  14. .. autosummary::
  15. :toctree: generated/
  16. :template: function.rst
  17. mobilenet_v2