diff --git a/torchvision/models/mobilenetv3.py b/torchvision/models/mobilenetv3.py index aa520e14962..1041d4d149f 100644 --- a/torchvision/models/mobilenetv3.py +++ b/torchvision/models/mobilenetv3.py @@ -378,7 +378,7 @@ def mobilenet_v3_large( weights are used. progress (bool, optional): If True, displays a progress bar of the download to stderr. Default is True. - **kwargs: parameters passed to the ``torchvision.models.resnet.MobileNetV3`` + **kwargs: parameters passed to the ``torchvision.models.mobilenet.MobileNetV3`` base class. Please refer to the `source code `_ for more details about this class. @@ -409,7 +409,7 @@ def mobilenet_v3_small( weights are used. progress (bool, optional): If True, displays a progress bar of the download to stderr. Default is True. - **kwargs: parameters passed to the ``torchvision.models.resnet.MobileNetV3`` + **kwargs: parameters passed to the ``torchvision.models.mobilenet.MobileNetV3`` base class. Please refer to the `source code `_ for more details about this class.