mnasnet.rst 686 B

12345678910111213141516171819202122232425262728
  1. MNASNet
  2. =======
  3. .. currentmodule:: torchvision.models
  4. The MNASNet model is based on the `MnasNet: Platform-Aware Neural Architecture
  5. Search for Mobile <https://arxiv.org/pdf/1807.11626.pdf>`__ paper.
  6. Model builders
  7. --------------
  8. The following model builders can be used to instantiate an MNASNet model.
  9. All the model builders internally rely on the
  10. ``torchvision.models.mnasnet.MNASNet`` base class. Please refer to the `source
  11. code
  12. <https://github.com/pytorch/vision/blob/main/torchvision/models/mnasnet.py>`_ for
  13. more details about this class.
  14. .. autosummary::
  15. :toctree: generated/
  16. :template: function.rst
  17. mnasnet0_5
  18. mnasnet0_75
  19. mnasnet1_0
  20. mnasnet1_3