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

docs: Added quantized ResNext to the new doc (#6032)



* docs: Added quantized ResNeXt to new docs

* docs: Fixed docstring
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent b50aaf0f
Quantized ResNeXt
=================
.. currentmodule:: torchvision.models.quantization
The quantized ResNext model is based on the `Aggregated Residual Transformations for Deep Neural Networks <https://arxiv.org/abs/1611.05431v2>`__
paper.
Model builders
--------------
The following model builders can be used to instantiate a quantized ResNeXt
model, with or without pre-trained weights. All the model builders internally
rely on the ``torchvision.models.quantization.resnet.QuantizableResNet``
base class. Please refer to the `source code
<https://github.com/pytorch/vision/blob/main/torchvision/models/quantization/resnet.py>`_
for more details about this class.
.. autosummary::
:toctree: generated/
:template: function.rst
resnext101_32x8d
resnext101_64x4d
...@@ -206,6 +206,7 @@ pre-trained weights: ...@@ -206,6 +206,7 @@ pre-trained weights:
models/mobilenetv2_quant models/mobilenetv2_quant
models/mobilenetv3_quant models/mobilenetv3_quant
models/resnet_quant models/resnet_quant
models/resnext_quant
models/shufflenetv2_quant models/shufflenetv2_quant
| |
......
...@@ -366,7 +366,7 @@ def resnext101_32x8d( ...@@ -366,7 +366,7 @@ def resnext101_32x8d(
**kwargs: Any, **kwargs: Any,
) -> QuantizableResNet: ) -> QuantizableResNet:
"""ResNeXt-101 32x8d model from """ResNeXt-101 32x8d model from
`Aggregated Residual Transformation for Deep Neural Networks <https://arxiv.org/abs/1611.05431.pdf>`_ `Aggregated Residual Transformation for Deep Neural Networks <https://arxiv.org/abs/1611.05431>`_
.. note:: .. note::
Note that ``quantize = True`` returns a quantized model with 8 bit Note that ``quantize = True`` returns a quantized model with 8 bit
...@@ -409,7 +409,7 @@ def resnext101_64x4d( ...@@ -409,7 +409,7 @@ def resnext101_64x4d(
**kwargs: Any, **kwargs: Any,
) -> QuantizableResNet: ) -> QuantizableResNet:
"""ResNeXt-101 64x4d model from """ResNeXt-101 64x4d model from
`Aggregated Residual Transformation for Deep Neural Networks <https://arxiv.org/abs/1611.05431.pdf>`_ `Aggregated Residual Transformation for Deep Neural Networks <https://arxiv.org/abs/1611.05431>`_
.. 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