We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 979b4d1 commit 51abcd4Copy full SHA for 51abcd4
graph_net/torch/extractor.py
@@ -11,7 +11,7 @@
11
def extract(name, dynamic=True, mut_graph_codes=None, placeholder_auto_rename=False):
12
"""
13
Extract computation graphs from PyTorch nn.Module.
14
- The extracted computation graph will be saved into directory of env var $GRAPH_NET_EXTRACT_WORKSPACE.
+ The extracted computation graphs will be saved into directory of env var $GRAPH_NET_EXTRACT_WORKSPACE.
15
16
Args:
17
name (str): The name of the model, used as the directory name for saving.
0 commit comments