Skip to content

Commit 651f604

Browse files
YosuaMichaelfacebook-github-bot
authored andcommitted
[fbsync] Fix alexnet docs (#5905)
Summary: Co-authored-by: Vasilis Vryniotis <[email protected]> Reviewed By: jdsgomes, NicolasHug Differential Revision: D36095712 fbshipit-source-id: f4e5d432da3b2ca53200224d6c05d7983b71c081
1 parent 7c06636 commit 651f604

File tree

2 files changed

+23
-22
lines changed

2 files changed

+23
-22
lines changed

docs/source/models/alexnet.rst

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,9 @@ AlexNet
33

44
.. currentmodule:: torchvision.models
55

6-
The AlexNet model is based on the `ImageNet Classification with Deep Convolutional Neural Networks
7-
<https://papers.nips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html>`__
6+
The AlexNet model is based on the
7+
`ImageNet Classification with Deep Convolutional Neural Networks
8+
<https://papers.nips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html>`__.
89
paper.
910

1011

torchvision/models/alexnet.py

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -72,26 +72,26 @@ class AlexNet_Weights(WeightsEnum):
7272

7373
@handle_legacy_interface(weights=("pretrained", AlexNet_Weights.IMAGENET1K_V1))
7474
def alexnet(*, weights: Optional[AlexNet_Weights] = None, progress: bool = True, **kwargs: Any) -> AlexNet:
75-
r"""AlexNet model architecture from the
76-
`ImageNet Classification with Deep Convolutional Neural Networks
77-
<https://papers.nips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html>`_ paper.
78-
The required minimum input size of the model is 63x63.
79-
80-
Args:
81-
weights (:class:`~torchvision.models.AlexNet_Weights`, optional): The
82-
pretrained weights to use. See
83-
:class:`~torchvision.models.AlexNet_Weights` below for
84-
more details, and possible values. By default, no pre-trained
85-
weights are used.
86-
progress (bool, optional): If True, displays a progress bar of the
87-
download to stderr. Default is True.
88-
**kwargs: parameters passed to the ``torchvision.models.squeezenet.AlexNet``
89-
base class. Please refer to the `source code
90-
<https://github.com/pytorch/vision/blob/main/torchvision/models/alexnet.py>`_
91-
for more details about this class.
92-
93-
.. autoclass:: torchvision.models.AlexNet_Weights
94-
:members:
75+
"""AlexNet model architecture from the `ImageNet Classification with Deep Convolutional Neural Networks
76+
<https://papers.nips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html>`__ paper.
77+
78+
The required minimum input size of the model is 63x63.
79+
80+
Args:
81+
weights (:class:`~torchvision.models.AlexNet_Weights`, optional): The
82+
pretrained weights to use. See
83+
:class:`~torchvision.models.AlexNet_Weights` below for
84+
more details, and possible values. By default, no pre-trained
85+
weights are used.
86+
progress (bool, optional): If True, displays a progress bar of the
87+
download to stderr. Default is True.
88+
**kwargs: parameters passed to the ``torchvision.models.squeezenet.AlexNet``
89+
base class. Please refer to the `source code
90+
<https://github.com/pytorch/vision/blob/main/torchvision/models/alexnet.py>`_
91+
for more details about this class.
92+
93+
.. autoclass:: torchvision.models.AlexNet_Weights
94+
:members:
9595
"""
9696

9797
weights = AlexNet_Weights.verify(weights)

0 commit comments

Comments
 (0)