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

Author: ooooo-create

graph_net/torch/utils.py

@@ -8,17 +8,23 @@
 import argparse
 import importlib
 import inspect
+import math
 
 
 def apply_templates(forward_code: str) -> str:
     tab = "    "
     forward_code = f"\n{tab}".join(forward_code.split("\n"))
-    return f"import torch\n\nclass GraphModule(torch.nn.Module):\n{tab}{forward_code}"
+    imports = "import torch"
+    if "device" in forward_code:
+        imports += "\n\nfrom torch import device"
+    return f"{imports}\n\nclass GraphModule(torch.nn.Module):\n{tab}{forward_code}"
 
 
 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}"
 
 

samples/timm/beit3_base_patch16_224.in22k_ft_in1k/graph_hash.txt

@@ -0,0 +1 @@
+49efabd40a93e856c10a3cf40aef7d7ad7e6f9fcfed3fc57f883ebd0e7d3f938

samples/timm/beit3_base_patch16_224.in22k_ft_in1k/graph_net.json

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