Merge branch 'develop' of github.com:PaddlePaddle/GraphNet into develop

Author: lixinqi

graph_net/torch/extractor.py

@@ -66,6 +66,7 @@ def try_rename_placeholder(node):
                 "framework": "torch",
                 "num_devices_required": 1,
                 "num_nodes_required": 1,
+                "dynamic": bool(dynamic),
             }
             with open(os.path.join(model_path, "graph_net.json"), "w") as f:
                 json.dump(metadata, f, indent=4)

graph_net/torch/utils.py

@@ -23,16 +23,19 @@ def apply_templates(forward_code: str) -> str:
 def get_limited_precision_float_str(value):
     if not isinstance(value, float):
         return value
-    if not math.isfinite(value):
-        return f'float("{value}")'
     return f"{value:.3f}"
 
 
 def convert_state_and_inputs_impl(state_dict, example_inputs):
     def tensor_info(tensor):
         is_float = tensor.dtype.is_floating_point
         mean = float(tensor.mean().item()) if is_float else None
-        std = float(tensor.std().item()) if is_float else None
+        std = None
+        if is_float:
+            if tensor.numel() <= 1:
+                std = 0.0
+            else:
+                std = float(tensor.std().item())
         return {
             "shape": list(tensor.shape),
             "dtype": str(tensor.dtype),

samples/transformers-auto-model/all-MiniLM-L6-v2/graph_hash.txt

@@ -0,0 +1 @@
+6d8b5686295201d8140f6155bc3bcc52982d4faa5003fdfb34ad62dc3ba93fd2

samples/transformers-auto-model/all-MiniLM-L6-v2/graph_net.json

@@ -0,0 +1,5 @@
+{
+    "framework": "torch",
+    "num_devices_required": 1,
+    "num_nodes_required": 1
+}