"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "d678357f06ff45eaf020f31c0147cd4c113347de"
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) ->
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:
......
......@@ -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:
......
......@@ -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:
......
......@@ -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:
......
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