mnasnet.rst 686 Bytes
Newer Older
Joao Gomes's avatar
Joao Gomes committed
1
2
3
4
5
6
7
8
9
10
11
12
13
MNASNet
=======

.. currentmodule:: torchvision.models


The MNASNet model is based on the `MnasNet: Platform-Aware Neural Architecture
Search for Mobile <https://arxiv.org/pdf/1807.11626.pdf>`__ paper.


Model builders
--------------

14
The following model builders can be used to instantiate an MNASNet model.
15
All the model builders internally rely on the
Joao Gomes's avatar
Joao Gomes committed
16
17
18
19
20
21
22
23
24
25
26
27
28
``torchvision.models.mnasnet.MNASNet`` base class. Please refer to the `source
code
<https://github.com/pytorch/vision/blob/main/torchvision/models/mnasnet.py>`_ for
more details about this class.

.. autosummary::
    :toctree: generated/
    :template: function.rst

    mnasnet0_5
    mnasnet0_75
    mnasnet1_0
    mnasnet1_3