Fix input dtype for paddle validate and enable counting paddle samples.

Author: Xreki

graph_net/paddle/utils.py

@@ -170,4 +170,4 @@ def replay_tensor(info):
     if "data" in info and info["data"] is not None:
         return info["data"].to(device)
 
-    return paddle.randn(shape).to(dtype).to(device) * std * 1e-3 + 1e-2
+    return (paddle.randn(shape).cast(dtype).to(device) * std * 1e-3 + 1e-2).cast(dtype)

graph_net/paddle/validate.py

@@ -71,10 +71,13 @@ def main(args):
     print(np.argmin(y), np.argmax(y))
     if isinstance(y, paddle.Tensor):
         print(y.shape)
-    elif isinstance(y, list) or isinstance(y, tuple):
+    elif (isinstance(y, list) or isinstance(y, tuple)) and all(
+        isinstance(obj, paddle.Tensor) for obj in y
+    ):
+        # list of paddle.Tensor
         print(y[0].shape)
     else:
-        raise ValueError("Illegal Return Value.")
+        raise ValueError("Illegal return value.")
 
     if not args.no_check_redundancy:
         print("Check redundancy ...")

tools/count_sample.py

@@ -4,21 +4,27 @@
 
 filename = os.path.abspath(__file__)
 root_dir = os.path.dirname(os.path.dirname(filename))
-samples_dir = os.path.join(root_dir, "samples")
-model_categories = os.listdir(samples_dir)
+framework2dirname = {
+    "torch": "samples",
+    "paddle": "paddle_samples",
+}
 
-graph_net_count = 0
-graph_net_dict = {}
-for category in model_categories:
-    category_dir = os.path.join(samples_dir, category)
-    if os.path.isdir(category_dir):
-        graph_net_dict[category] = 0
-        for root, dirs, files in os.walk(category_dir):
-            if "graph_net.json" in files:
-                graph_net_count += 1
-                graph_net_dict[category] += 1
+for framework in ["torch", "paddle"]:
+    samples_dir = os.path.join(root_dir, framework2dirname[framework])
+    model_categories = os.listdir(samples_dir)
 
-print(f"Number of graph_net.json files: {graph_net_count}")
-for name, number in graph_net_dict.items():
-    print(f"- {name:24}: {number}")
-print()
+    graph_net_count = 0
+    graph_net_dict = {}
+    for category in model_categories:
+        category_dir = os.path.join(samples_dir, category)
+        if os.path.isdir(category_dir):
+            graph_net_dict[category] = 0
+            for root, dirs, files in os.walk(category_dir):
+                if "graph_net.json" in files:
+                    graph_net_count += 1
+                    graph_net_dict[category] += 1
+
+    print(f"Number of {framework} samples: {graph_net_count}")
+    for name, number in graph_net_dict.items():
+        print(f"- {name:24}: {number}")
+    print()