diff --git a/torchvision/models/quantization/resnet.py b/torchvision/models/quantization/resnet.py index 70dd5d92f0e..5af87a1e586 100644 --- a/torchvision/models/quantization/resnet.py +++ b/torchvision/models/quantization/resnet.py @@ -378,7 +378,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 @@ -391,10 +391,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: """ @@ -421,7 +421,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 @@ -434,10 +434,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: """