diff --git a/docs/source/models/resnext_quant.rst b/docs/source/models/resnext_quant.rst new file mode 100644 index 00000000000..916b9e4a39a --- /dev/null +++ b/docs/source/models/resnext_quant.rst @@ -0,0 +1,25 @@ +Quantized ResNeXt +================= + +.. currentmodule:: torchvision.models.quantization + +The quantized ResNext model is based on the `Aggregated Residual Transformations for Deep Neural Networks `__ +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 +`_ +for more details about this class. + +.. autosummary:: + :toctree: generated/ + :template: function.rst + + resnext101_32x8d + resnext101_64x4d diff --git a/docs/source/models_new.rst b/docs/source/models_new.rst index 6ded8480425..94ffe733538 100644 --- a/docs/source/models_new.rst +++ b/docs/source/models_new.rst @@ -206,6 +206,7 @@ pre-trained weights: models/mobilenetv2_quant models/mobilenetv3_quant models/resnet_quant + models/resnext_quant models/shufflenetv2_quant | diff --git a/torchvision/models/quantization/resnet.py b/torchvision/models/quantization/resnet.py index 65552f3a16a..dc3ee4c35c5 100644 --- a/torchvision/models/quantization/resnet.py +++ b/torchvision/models/quantization/resnet.py @@ -366,7 +366,7 @@ def resnext101_32x8d( **kwargs: Any, ) -> QuantizableResNet: """ResNeXt-101 32x8d model from - `Aggregated Residual Transformation for Deep Neural Networks `_ + `Aggregated Residual Transformation for Deep Neural Networks `_ .. note:: Note that ``quantize = True`` returns a quantized model with 8 bit @@ -409,7 +409,7 @@ def resnext101_64x4d( **kwargs: Any, ) -> QuantizableResNet: """ResNeXt-101 64x4d model from - `Aggregated Residual Transformation for Deep Neural Networks `_ + `Aggregated Residual Transformation for Deep Neural Networks `_ .. note:: Note that ``quantize = True`` returns a quantized model with 8 bit