diff --git a/torchvision/models/googlenet.py b/torchvision/models/googlenet.py index 7ead46a4ef6..476da0d8a53 100644 --- a/torchvision/models/googlenet.py +++ b/torchvision/models/googlenet.py @@ -287,7 +287,7 @@ def googlenet(pretrained: bool = False, progress: bool = True, **kwargs: Any) -> aux_logits (bool): If True, adds two auxiliary branches that can improve training. Default: *False* when pretrained is True otherwise *True* transform_input (bool): If True, preprocesses the input according to the method with which it - was trained on ImageNet. Default: *False* + was trained on ImageNet. Default: True if ``pretrained=True``, else False. """ if pretrained: if "transform_input" not in kwargs: diff --git a/torchvision/models/inception.py b/torchvision/models/inception.py index 714e5e9e406..85236c6dfb3 100644 --- a/torchvision/models/inception.py +++ b/torchvision/models/inception.py @@ -422,7 +422,7 @@ def inception_v3(pretrained: bool = False, progress: bool = True, **kwargs: Any) aux_logits (bool): If True, add an auxiliary branch that can improve training. Default: *True* transform_input (bool): If True, preprocesses the input according to the method with which it - was trained on ImageNet. Default: *False* + was trained on ImageNet. Default: True if ``pretrained=True``, else False. """ if pretrained: if "transform_input" not in kwargs: diff --git a/torchvision/models/quantization/googlenet.py b/torchvision/models/quantization/googlenet.py index d81f227b6f1..ff2352daa35 100644 --- a/torchvision/models/quantization/googlenet.py +++ b/torchvision/models/quantization/googlenet.py @@ -123,7 +123,7 @@ def googlenet( aux_logits (bool): If True, adds two auxiliary branches that can improve training. Default: *False* when pretrained is True otherwise *True* transform_input (bool): If True, preprocesses the input according to the method with which it - was trained on ImageNet. Default: *False* + was trained on ImageNet. Default: True if ``pretrained=True``, else False. """ if pretrained: if "transform_input" not in kwargs: diff --git a/torchvision/models/quantization/inception.py b/torchvision/models/quantization/inception.py index 83665be370f..379f4da548e 100644 --- a/torchvision/models/quantization/inception.py +++ b/torchvision/models/quantization/inception.py @@ -197,7 +197,7 @@ def inception_v3( aux_logits (bool): If True, add an auxiliary branch that can improve training. Default: *True* transform_input (bool): If True, preprocesses the input according to the method with which it - was trained on ImageNet. Default: *False* + was trained on ImageNet. Default: True if ``pretrained=True``, else False. """ if pretrained: if "transform_input" not in kwargs: