Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 23 additions and 22 deletions
docs/source/models/alexnet.rst
View file @ 36c85c79
...@@ -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.
......
torchvision/models/alexnet.py
View file @ 36c85c79
...@@ -72,9 +72,9 @@ class AlexNet_Weights(WeightsEnum): ...@@ -72,9 +72,9 @@ 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:
......
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