Skip to content

GitLab

Unverified Commit b50aaf0f authored by F-G Fernandez's avatar F-G Fernandez Committed by GitHub
Browse files

docs: Fixed quantized resnet docstring (#6033)

parent a1232c21
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/models/quantization/resnet.py
View file @ b50aaf0f
...@@ -268,7 +268,7 @@ def resnet18( ...@@ -268,7 +268,7 @@ def resnet18(
**kwargs: Any, **kwargs: Any,
) -> QuantizableResNet: ) -> QuantizableResNet:
"""ResNet-18 model from """ResNet-18 model from
`Deep Residual Learning for Image Recognition <https://arxiv.org/abs/1512.03385.pdf>`_ `Deep Residual Learning for Image Recognition <https://arxiv.org/abs/1512.03385>`_
.. note:: .. note::
Note that ``quantize = True`` returns a quantized model with 8 bit Note that ``quantize = True`` returns a quantized model with 8 bit
...@@ -317,7 +317,7 @@ def resnet50( ...@@ -317,7 +317,7 @@ def resnet50(
**kwargs: Any, **kwargs: Any,
) -> QuantizableResNet: ) -> QuantizableResNet:
"""ResNet-50 model from """ResNet-50 model from
`Deep Residual Learning for Image Recognition <https://arxiv.org/abs/1512.03385.pdf>`_ `Deep Residual Learning for Image Recognition <https://arxiv.org/abs/1512.03385>`_
.. note:: .. note::
Note that ``quantize = True`` returns a quantized model with 8 bit Note that ``quantize = True`` returns a quantized model with 8 bit
......
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