"...text-generation-inference.git" did not exist on "c2fd35d875155d858a60542edabb9df59587e1f8"
Unverified Commit 36c85c79 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix alexnet docs (#5905)


Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent b53c91d0
...@@ -3,8 +3,9 @@ AlexNet ...@@ -3,8 +3,9 @@ AlexNet
.. currentmodule:: torchvision.models .. currentmodule:: torchvision.models
The AlexNet model is based on the `ImageNet Classification with Deep Convolutional Neural Networks The AlexNet model is based on the
<https://papers.nips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html>`__ `ImageNet Classification with Deep Convolutional Neural Networks
<https://papers.nips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html>`__.
paper. paper.
......
...@@ -72,26 +72,26 @@ class AlexNet_Weights(WeightsEnum): ...@@ -72,26 +72,26 @@ class AlexNet_Weights(WeightsEnum):
@handle_legacy_interface(weights=("pretrained", AlexNet_Weights.IMAGENET1K_V1)) @handle_legacy_interface(weights=("pretrained", AlexNet_Weights.IMAGENET1K_V1))
def alexnet(*, weights: Optional[AlexNet_Weights] = None, progress: bool = True, **kwargs: Any) -> AlexNet: def alexnet(*, weights: Optional[AlexNet_Weights] = None, progress: bool = True, **kwargs: Any) -> AlexNet:
r"""AlexNet model architecture from the """AlexNet model architecture from the `ImageNet Classification with Deep Convolutional Neural Networks
`ImageNet Classification with Deep Convolutional Neural Networks <https://papers.nips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html>`__ paper.
<https://papers.nips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html>`_ paper.
The required minimum input size of the model is 63x63. The required minimum input size of the model is 63x63.
Args: Args:
weights (:class:`~torchvision.models.AlexNet_Weights`, optional): The weights (:class:`~torchvision.models.AlexNet_Weights`, optional): The
pretrained weights to use. See pretrained weights to use. See
:class:`~torchvision.models.AlexNet_Weights` below for :class:`~torchvision.models.AlexNet_Weights` below for
more details, and possible values. By default, no pre-trained 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.
**kwargs: parameters passed to the ``torchvision.models.squeezenet.AlexNet`` **kwargs: parameters passed to the ``torchvision.models.squeezenet.AlexNet``
base class. Please refer to the `source code base class. Please refer to the `source code
<https://github.com/pytorch/vision/blob/main/torchvision/models/alexnet.py>`_ <https://github.com/pytorch/vision/blob/main/torchvision/models/alexnet.py>`_
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.AlexNet_Weights .. autoclass:: torchvision.models.AlexNet_Weights
:members: :members:
""" """
weights = AlexNet_Weights.verify(weights) weights = AlexNet_Weights.verify(weights)
......
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