Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
b8b2294e
"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "d678357f06ff45eaf020f31c0147cd4c113347de"
Unverified
Commit
b8b2294e
authored
Dec 13, 2021
by
Nicolas Hug
Committed by
GitHub
Dec 13, 2021
Browse files
Fix googlenet and inception docstring for transform_input parameter (#5089)
parent
6a1d9ee7
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
4 additions
and
4 deletions
+4
-4
torchvision/models/googlenet.py
torchvision/models/googlenet.py
+1
-1
torchvision/models/inception.py
torchvision/models/inception.py
+1
-1
torchvision/models/quantization/googlenet.py
torchvision/models/quantization/googlenet.py
+1
-1
torchvision/models/quantization/inception.py
torchvision/models/quantization/inception.py
+1
-1
No files found.
torchvision/models/googlenet.py
View file @
b8b2294e
...
...
@@ -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
:
...
...
torchvision/models/inception.py
View file @
b8b2294e
...
...
@@ -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
:
...
...
torchvision/models/quantization/googlenet.py
View file @
b8b2294e
...
...
@@ -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
:
...
...
torchvision/models/quantization/inception.py
View file @
b8b2294e
...
...
@@ -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
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment