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 5ba2b8b commit 2b241f0Copy full SHA for 2b241f0
graph_net/torch/extractor.py
@@ -4,6 +4,8 @@
4
from typing import Union, Callable
5
from . import utils
6
7
+torch._dynamo.config.capture_scalar_outputs = True
8
+
9
10
def extract(name, dynamic=True, mut_graph_codes=None, placeholder_auto_rename=False):
11
"""
0 commit comments