diff --git a/docs/source/models/mnasnet.rst b/docs/source/models/mnasnet.rst index b4239be5c27..f902bc2ef38 100644 --- a/docs/source/models/mnasnet.rst +++ b/docs/source/models/mnasnet.rst @@ -11,8 +11,9 @@ Search for Mobile `__ 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 `_ for diff --git a/torchvision/models/mnasnet.py b/torchvision/models/mnasnet.py index 7e77cb90a3a..7f4751f12a3 100644 --- a/torchvision/models/mnasnet.py +++ b/torchvision/models/mnasnet.py @@ -306,10 +306,8 @@ def mnasnet0_75(*, weights: Optional[MNASNet0_75_Weights] = None, progress: bool `_ 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 = `_ 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.