Unverified Commit 22400011 authored by Joao Gomes's avatar Joao Gomes Committed by GitHub
Browse files

Remove references to non-existing weights in MNASNET (#5936)



* Remove references to non-existing weights

* Update torchvision/models/mnasnet.py
Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>

* Update torchvision/models/mnasnet.py
Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>

* apply ufmt
Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
parent 0a94ce70
......@@ -11,8 +11,9 @@ Search for Mobile <https://arxiv.org/pdf/1807.11626.pdf>`__ paper.
Model builders
--------------
The following model builders can be used to instanciate an MNASNet model, with or
without pre-trained weights. All the model builders internally rely on the
The following model builders can be used to instanciate an MNASNet model. Currently
only ``mnasnet0_5`` and ``mnasnet1_0`` can be instantiated with pre-trained weights.
All the model builders internally rely on the
``torchvision.models.mnasnet.MNASNet`` base class. Please refer to the `source
code
<https://github.com/pytorch/vision/blob/main/torchvision/models/mnasnet.py>`_ for
......
......@@ -306,10 +306,8 @@ def mnasnet0_75(*, weights: Optional[MNASNet0_75_Weights] = None, progress: bool
<https://arxiv.org/pdf/1807.11626.pdf>`_ paper.
Args:
weights (:class:`~torchvision.models.MNASNet0_75_Weights`, optional): The
pretrained weights to use. See
:class:`~torchvision.models.MNASNet0_75_Weights` below for
more details, and possible values. By default, no pre-trained
weights (:class:`~torchvision.models.MNASNet0_75_Weights`, optional): Currently
no pre-trained weights are available and by default no pre-trained
weights are used.
progress (bool, optional): If True, displays a progress bar of the
download to stderr. Default is True.
......@@ -358,10 +356,8 @@ def mnasnet1_3(*, weights: Optional[MNASNet1_3_Weights] = None, progress: bool =
<https://arxiv.org/pdf/1807.11626.pdf>`_ paper.
Args:
weights (:class:`~torchvision.models.MNASNet1_3_Weights`, optional): The
pretrained weights to use. See
:class:`~torchvision.models.MNASNet1_3_Weights` below for
more details, and possible values. By default, no pre-trained
weights (:class:`~torchvision.models.MNASNet1_3_Weights`, optional): Currently
no pre-trained weights are available and by default no pre-trained
weights are used.
progress (bool, optional): If True, displays a progress bar of the
download to stderr. Default is True.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment