fix

Author: fangfangssj

graph_net/torch/extractor.py

@@ -5,6 +5,8 @@
 from . import utils
 
 torch._dynamo.config.capture_scalar_outputs = True
+torch._dynamo.capture_dynamic_output_shape_ops = True
+torch._dynamo.config.suppress_errors = True
 
 
 def extract(name, dynamic=True, mut_graph_codes=None, placeholder_auto_rename=False):