From 713f7ea9ce3242182e5f8ea40545fb63c43487ab Mon Sep 17 00:00:00 2001 From: Kazuki Adachi Date: Mon, 18 Apr 2022 20:07:45 +0900 Subject: [PATCH] Update feature_extraction.py --- torchvision/models/feature_extraction.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/models/feature_extraction.py b/torchvision/models/feature_extraction.py index 8a25e1f7187..1b380076b2a 100644 --- a/torchvision/models/feature_extraction.py +++ b/torchvision/models/feature_extraction.py @@ -226,7 +226,7 @@ def get_graph_node_names( Args: 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 `torch.fx.Tracer `_). By default it will be set to wrap and make leaf nodes all torchvision ops: @@ -389,7 +389,7 @@ def create_feature_extractor( for train mode are different than those from eval mode. If this is specified, ``train_return_nodes`` must also 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 `torch.fx.Tracer `_). By default it will be set to wrap and make leaf nodes all torchvision ops: