File tree Expand file tree Collapse file tree 3 files changed +28
-2
lines changed
torchvision/models/quantization Expand file tree Collapse file tree 3 files changed +28
-2
lines changed Original file line number Diff line number Diff line change 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
Original file line number Diff line number Diff 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|
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments