Fix a sample.

Author: Xreki

graph_net/subgraph_decompose_and_evaluation_step.py

@@ -301,19 +301,14 @@ def generate_refined_tasks(base_output_dir, current_pass_id):
     print(f"[Init] Resuming from Pass_{current_pass_id - 1} (Dir: {prev_pass_dir})...")
 
     prev_config = DecomposeConfig.load(prev_pass_dir)
-    prev_incorrect_subgraphs = prev_config.incorrect_models
-    prev_tasks_map = prev_config.tasks_map
-    running_states = prev_config.running_states
-
-    # Load previous max size as fallback
-    prev_max_subgraph_size = prev_config.max_subgraph_size
-    max_subgraph_size = prev_max_subgraph_size // 2
-
-    if not prev_incorrect_subgraphs:
-        return {}, max_subgraph_size, running_states
+    max_subgraph_size = prev_config.max_subgraph_size // 2
+    if not prev_config.incorrect_models:
+        return {}, max_subgraph_size, prev_config.running_states
 
     tasks_map = {}
-    for subgraph_path in prev_incorrect_subgraphs:
+    prev_tasks_map = prev_config.tasks_map
+
+    for subgraph_path in prev_config.incorrect_models:
         # Parse model name and subgraph index
         model_name_with_subgraph_idx = subgraph_path.rstrip("/").split(os.sep)[-1]
         model_name = "_".join(model_name_with_subgraph_idx.split("_")[:-1])
@@ -336,7 +331,7 @@ def generate_refined_tasks(base_output_dir, current_pass_id):
                 "split_positions": set(),
             }
 
-    return tasks_map, max_subgraph_size, running_states
+    return tasks_map, max_subgraph_size, prev_config.running_states
 
 
 def prepare_tasks_and_verify(args, current_pass_id, base_output_dir):

paddle_samples/PaddleX/Mask-RT-DETR-S/subgraph_14/input_meta.py

@@ -217,9 +217,7 @@ class Program_weight_tensor_data_21:
     shape = [1, 8400, 4]
     dtype = "float32"
     min_val = float("-4.36945")
-    max_val = float("inf")
-    mean = float("inf")
-    std = float("nan")
+    max_val = float("5.0")
     data = None