Unverified Commit 53c85328 authored by Kazuki Adachi's avatar Kazuki Adachi Committed by GitHub
Browse files

Typo fixes (#5827)

parent 10bb5b18
...@@ -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 keywork arguments for tracer_kwargs (dict, optional): a dictionary of keyword 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 keywork arguments for tracer_kwargs (dict, optional): a dictionary of keyword 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:
......
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