Skip to content

Commit a068746

Browse files
YosuaMichaelfacebook-github-bot
authored andcommitted
[fbsync] Typo fixes (#5827)
Reviewed By: jdsgomes, NicolasHug Differential Revision: D36095687 fbshipit-source-id: 8d10ed59586f181be1562ec791116f51c2e50403
1 parent bbc3b25 commit a068746

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchvision/models/feature_extraction.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ def get_graph_node_names(
226226
227227
Args:
228228
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
230230
``NodePathTracer`` (they are eventually passed onto
231231
`torch.fx.Tracer <https://pytorch.org/docs/stable/fx.html#torch.fx.Tracer>`_).
232232
By default it will be set to wrap and make leaf nodes all torchvision ops:
@@ -389,7 +389,7 @@ def create_feature_extractor(
389389
for train mode are different than those from eval mode.
390390
If this is specified, ``train_return_nodes`` must also be specified,
391391
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
393393
``NodePathTracer`` (which passes them onto it's parent class
394394
`torch.fx.Tracer <https://pytorch.org/docs/stable/fx.html#torch.fx.Tracer>`_).
395395
By default it will be set to wrap and make leaf nodes all torchvision ops:

0 commit comments

Comments
 (0)