[Bug Fix] new sample STFPM to validate fix_hash

graph_net/paddle/validate.py

@@ -12,6 +12,7 @@
 import graph_net
 import os
 import ast
+import astor
 import paddle
 
 
@@ -29,7 +30,7 @@ def _get_sha_hash(content):
     return m.hexdigest()
 
 
-def _extract_forward_source(model_path):
+def _extract_forward_source(model_path, class_name):
     source = None
     with open(f"{model_path}/model.py", "r") as f:
         source = f.read()
@@ -38,18 +39,18 @@ def _extract_forward_source(model_path):
     forward_code = None
 
     for node in tree.body:
-        if isinstance(node, ast.ClassDef) and node.name == "GraphModule":
+        if isinstance(node, ast.ClassDef) and node.name == class_name:
             for fn in node.body:
                 if isinstance(fn, ast.FunctionDef) and fn.name == "forward":
-                    return ast.unparse(fn)
+                    return astor.to_source(fn)
     return None
 
 
 def check_graph_hash(args):
     model_path = args.model_path
     file_path = f"{model_path}/graph_hash.txt"
     if args.dump_graph_hash_key:
-        model_str = _extract_forward_source(model_path)
+        model_str = _extract_forward_source(model_path, class_name="GraphModule")
         assert model_str is not None, f"model_str of {args.model_path} is None."
         new_hash_text = _get_sha_hash(model_str)
 

paddle_samples/PaddleX/STFPM/graph_hash.txt

@@ -0,0 +1 @@
+32db5c97231520cb0081bc48365b4cb65eeda8c2a860564cbce9488f14392121

paddle_samples/PaddleX/STFPM/graph_net.json

@@ -0,0 +1,6 @@
+{
+    "framework": "paddle",
+    "model_name": "STFPM",
+    "num_devices_required": 1,
+    "num_nodes_required": 1
+}

paddle_samples/PaddleX/STFPM/input_meta.py

@@ -0,0 +1,9 @@
+class Program_weight_tensor_data_0:
+    name = "data_0"
+    shape = [1, 3, 256, 256]
+    dtype = "float32"
+    min_val = float("-1.54422")
+    max_val = float("1.21516")
+    mean = float("-0.201062")
+    std = float("0.553945")
+    data = None