File tree 1 file changed +2
-2
lines changed 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -226,7 +226,7 @@ def get_graph_node_names(
226
226
227
227
Args:
228
228
model (nn.Module): model for which we'd like to print node names
229
- tracer_kwargs (dict, optional): a dictionary of keywork arguments for
229
+ tracer_kwargs (dict, optional): a dictionary of keyword arguments for
230
230
``NodePathTracer`` (they are eventually passed onto
231
231
`torch.fx.Tracer <https://pytorch.org/docs/stable/fx.html#torch.fx.Tracer>`_).
232
232
By default it will be set to wrap and make leaf nodes all torchvision ops:
@@ -389,7 +389,7 @@ def create_feature_extractor(
389
389
for train mode are different than those from eval mode.
390
390
If this is specified, ``train_return_nodes`` must also be specified,
391
391
and `return_nodes` should not be specified.
392
- tracer_kwargs (dict, optional): a dictionary of keywork arguments for
392
+ tracer_kwargs (dict, optional): a dictionary of keyword arguments for
393
393
``NodePathTracer`` (which passes them onto it's parent class
394
394
`torch.fx.Tracer <https://pytorch.org/docs/stable/fx.html#torch.fx.Tracer>`_).
395
395
By default it will be set to wrap and make leaf nodes all torchvision ops:
You can’t perform that action at this time.
0 commit comments