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
53c85328
Unverified
Commit
53c85328
authored
Apr 19, 2022
by
Kazuki Adachi
Committed by
GitHub
Apr 18, 2022
Browse files
Typo fixes (#5827)
parent
10bb5b18
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
torchvision/models/feature_extraction.py
torchvision/models/feature_extraction.py
+2
-2
No files found.
torchvision/models/feature_extraction.py
View file @
53c85328
...
@@ -226,7 +226,7 @@ def get_graph_node_names(
...
@@ -226,7 +226,7 @@ def get_graph_node_names(
Args:
Args:
model (nn.Module): model for which we'd like to print node names
model (nn.Module): model for which we'd like to print node names
tracer_kwargs (dict, optional): a dictionary of keywor
k
arguments for
tracer_kwargs (dict, optional): a dictionary of keywor
d
arguments for
``NodePathTracer`` (they are eventually passed onto
``NodePathTracer`` (they are eventually passed onto
`torch.fx.Tracer <https://pytorch.org/docs/stable/fx.html#torch.fx.Tracer>`_).
`torch.fx.Tracer <https://pytorch.org/docs/stable/fx.html#torch.fx.Tracer>`_).
By default it will be set to wrap and make leaf nodes all torchvision ops:
By default it will be set to wrap and make leaf nodes all torchvision ops:
...
@@ -389,7 +389,7 @@ def create_feature_extractor(
...
@@ -389,7 +389,7 @@ def create_feature_extractor(
for train mode are different than those from eval mode.
for train mode are different than those from eval mode.
If this is specified, ``train_return_nodes`` must also be specified,
If this is specified, ``train_return_nodes`` must also be specified,
and `return_nodes` should not be specified.
and `return_nodes` should not be specified.
tracer_kwargs (dict, optional): a dictionary of keywor
k
arguments for
tracer_kwargs (dict, optional): a dictionary of keywor
d
arguments for
``NodePathTracer`` (which passes them onto it's parent class
``NodePathTracer`` (which passes them onto it's parent class
`torch.fx.Tracer <https://pytorch.org/docs/stable/fx.html#torch.fx.Tracer>`_).
`torch.fx.Tracer <https://pytorch.org/docs/stable/fx.html#torch.fx.Tracer>`_).
By default it will be set to wrap and make leaf nodes all torchvision ops:
By default it will be set to wrap and make leaf nodes all torchvision ops:
...
...
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