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. ...@@ -11,8 +11,9 @@ Search for Mobile <https://arxiv.org/pdf/1807.11626.pdf>`__ paper.
Model builders Model builders
-------------- --------------
The following model builders can be used to instanciate an MNASNet model, with or The following model builders can be used to instanciate an MNASNet model. Currently
without pre-trained weights. All the model builders internally rely on the 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 ``torchvision.models.mnasnet.MNASNet`` base class. Please refer to the `source
code code
<https://github.com/pytorch/vision/blob/main/torchvision/models/mnasnet.py>`_ for <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 ...@@ -306,10 +306,8 @@ def mnasnet0_75(*, weights: Optional[MNASNet0_75_Weights] = None, progress: bool
<https://arxiv.org/pdf/1807.11626.pdf>`_ paper. <https://arxiv.org/pdf/1807.11626.pdf>`_ paper.
Args: Args:
weights (:class:`~torchvision.models.MNASNet0_75_Weights`, optional): The weights (:class:`~torchvision.models.MNASNet0_75_Weights`, optional): Currently
pretrained weights to use. See no pre-trained weights are available and by default no pre-trained
:class:`~torchvision.models.MNASNet0_75_Weights` below for
more details, and possible values. By default, no pre-trained
weights are used. weights are used.
progress (bool, optional): If True, displays a progress bar of the progress (bool, optional): If True, displays a progress bar of the
download to stderr. Default is True. download to stderr. Default is True.
...@@ -358,10 +356,8 @@ def mnasnet1_3(*, weights: Optional[MNASNet1_3_Weights] = None, progress: bool = ...@@ -358,10 +356,8 @@ def mnasnet1_3(*, weights: Optional[MNASNet1_3_Weights] = None, progress: bool =
<https://arxiv.org/pdf/1807.11626.pdf>`_ paper. <https://arxiv.org/pdf/1807.11626.pdf>`_ paper.
Args: Args:
weights (:class:`~torchvision.models.MNASNet1_3_Weights`, optional): The weights (:class:`~torchvision.models.MNASNet1_3_Weights`, optional): Currently
pretrained weights to use. See no pre-trained weights are available and by default no pre-trained
:class:`~torchvision.models.MNASNet1_3_Weights` below for
more details, and possible values. By default, no pre-trained
weights are used. weights are used.
progress (bool, optional): If True, displays a progress bar of the progress (bool, optional): If True, displays a progress bar of the
download to stderr. Default is True. 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