Skip to content

Commit 10acc82

Browse files
frgfmdatumbox
andauthored
docs: Added quantized ResNext to the new doc (#6032)
* docs: Added quantized ResNeXt to new docs * docs: Fixed docstring Co-authored-by: Vasilis Vryniotis <[email protected]>
1 parent b50aaf0 commit 10acc82

File tree

3 files changed

+28
-2
lines changed

3 files changed

+28
-2
lines changed
Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
Quantized ResNeXt
2+
=================
3+
4+
.. currentmodule:: torchvision.models.quantization
5+
6+
The quantized ResNext model is based on the `Aggregated Residual Transformations for Deep Neural Networks <https://arxiv.org/abs/1611.05431v2>`__
7+
paper.
8+
9+
10+
Model builders
11+
--------------
12+
13+
The following model builders can be used to instantiate a quantized ResNeXt
14+
model, with or without pre-trained weights. All the model builders internally
15+
rely on the ``torchvision.models.quantization.resnet.QuantizableResNet``
16+
base class. Please refer to the `source code
17+
<https://github.com/pytorch/vision/blob/main/torchvision/models/quantization/resnet.py>`_
18+
for more details about this class.
19+
20+
.. autosummary::
21+
:toctree: generated/
22+
:template: function.rst
23+
24+
resnext101_32x8d
25+
resnext101_64x4d

docs/source/models_new.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -206,6 +206,7 @@ pre-trained weights:
206206
models/mobilenetv2_quant
207207
models/mobilenetv3_quant
208208
models/resnet_quant
209+
models/resnext_quant
209210
models/shufflenetv2_quant
210211

211212
|

torchvision/models/quantization/resnet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,7 @@ def resnext101_32x8d(
366366
**kwargs: Any,
367367
) -> QuantizableResNet:
368368
"""ResNeXt-101 32x8d model from
369-
`Aggregated Residual Transformation for Deep Neural Networks <https://arxiv.org/abs/1611.05431.pdf>`_
369+
`Aggregated Residual Transformation for Deep Neural Networks <https://arxiv.org/abs/1611.05431>`_
370370
371371
.. note::
372372
Note that ``quantize = True`` returns a quantized model with 8 bit
@@ -409,7 +409,7 @@ def resnext101_64x4d(
409409
**kwargs: Any,
410410
) -> QuantizableResNet:
411411
"""ResNeXt-101 64x4d model from
412-
`Aggregated Residual Transformation for Deep Neural Networks <https://arxiv.org/abs/1611.05431.pdf>`_
412+
`Aggregated Residual Transformation for Deep Neural Networks <https://arxiv.org/abs/1611.05431>`_
413413
414414
.. note::
415415
Note that ``quantize = True`` returns a quantized model with 8 bit

0 commit comments

Comments
 (0)