From bd4d5c2a6ec8dff352811f87d40c5ad3a79c414b Mon Sep 17 00:00:00 2001 From: Aditya Oke Date: Wed, 18 May 2022 16:14:58 +0530 Subject: [PATCH] Fix resnext docs --- torchvision/models/quantization/resnet.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/torchvision/models/quantization/resnet.py b/torchvision/models/quantization/resnet.py index dc3ee4c35c5..8792263bf70 100644 --- a/torchvision/models/quantization/resnet.py +++ b/torchvision/models/quantization/resnet.py @@ -374,7 +374,7 @@ def resnext101_32x8d( GPU inference is not yet supported. Args: - weights (:class:`~torchvision.models.quantization.ResNet101_32X8D_QuantizedWeights` or :class:`~torchvision.models.ResNet101_32X8D_Weights`, optional): The + weights (:class:`~torchvision.models.quantization.ResNeXt101_32X8D_QuantizedWeights` or :class:`~torchvision.models.ResNeXt101_32X8D_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.ResNet101_32X8D_QuantizedWeights` below for more details, and possible values. By default, no pre-trained @@ -387,10 +387,10 @@ def resnext101_32x8d( `_ for more details about this class. - .. autoclass:: torchvision.models.quantization.ResNet101_32X8D_QuantizedWeights + .. autoclass:: torchvision.models.quantization.ResNeXt101_32X8D_QuantizedWeights :members: - .. autoclass:: torchvision.models.ResNet101_32X8D_Weights + .. autoclass:: torchvision.models.ResNeXt101_32X8D_Weights :members: :noindex: """ @@ -417,7 +417,7 @@ def resnext101_64x4d( GPU inference is not yet supported. Args: - weights (:class:`~torchvision.models.quantization.ResNet101_64X4D_QuantizedWeights` or :class:`~torchvision.models.ResNet101_64X4D_Weights`, optional): The + weights (:class:`~torchvision.models.quantization.ResNeXt101_64X4D_QuantizedWeights` or :class:`~torchvision.models.ResNeXt101_64X4D_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.ResNet101_64X4D_QuantizedWeights` below for more details, and possible values. By default, no pre-trained @@ -430,10 +430,10 @@ def resnext101_64x4d( `_ for more details about this class. - .. autoclass:: torchvision.models.quantization.ResNet101_64X4D_QuantizedWeights + .. autoclass:: torchvision.models.quantization.ResNeXt101_64X4D_QuantizedWeights :members: - .. autoclass:: torchvision.models.ResNet101_64X4D_Weights + .. autoclass:: torchvision.models.ResNeXt101_64X4D_Weights :members: :noindex: """