Unverified Commit b8b2294e authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix googlenet and inception docstring for transform_input parameter (#5089)

parent 6a1d9ee7
...@@ -287,7 +287,7 @@ def googlenet(pretrained: bool = False, progress: bool = True, **kwargs: Any) -> ...@@ -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. aux_logits (bool): If True, adds two auxiliary branches that can improve training.
Default: *False* when pretrained is True otherwise *True* Default: *False* when pretrained is True otherwise *True*
transform_input (bool): If True, preprocesses the input according to the method with which it 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 pretrained:
if "transform_input" not in kwargs: if "transform_input" not in kwargs:
......
...@@ -422,7 +422,7 @@ def inception_v3(pretrained: bool = False, progress: bool = True, **kwargs: Any) ...@@ -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. aux_logits (bool): If True, add an auxiliary branch that can improve training.
Default: *True* Default: *True*
transform_input (bool): If True, preprocesses the input according to the method with which it 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 pretrained:
if "transform_input" not in kwargs: if "transform_input" not in kwargs:
......
...@@ -123,7 +123,7 @@ def googlenet( ...@@ -123,7 +123,7 @@ def googlenet(
aux_logits (bool): If True, adds two auxiliary branches that can improve training. aux_logits (bool): If True, adds two auxiliary branches that can improve training.
Default: *False* when pretrained is True otherwise *True* Default: *False* when pretrained is True otherwise *True*
transform_input (bool): If True, preprocesses the input according to the method with which it 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 pretrained:
if "transform_input" not in kwargs: if "transform_input" not in kwargs:
......
...@@ -197,7 +197,7 @@ def inception_v3( ...@@ -197,7 +197,7 @@ def inception_v3(
aux_logits (bool): If True, add an auxiliary branch that can improve training. aux_logits (bool): If True, add an auxiliary branch that can improve training.
Default: *True* Default: *True*
transform_input (bool): If True, preprocesses the input according to the method with which it 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 pretrained:
if "transform_input" not in kwargs: if "transform_input" not in kwargs:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment