support model-path-prefix in splitting positions

Author: JewelRoam

graph_net/test/typical_sequence_decomposer_test.sh

@@ -11,6 +11,7 @@ model_list="$GRAPH_NET_ROOT/graph_net/test/dev_model_list/validation_error_model
 
 python3 -m graph_net.torch.typical_sequence_split_points \
     --model-list "$model_list" \
+    --model-path-prefix "$GRAPH_NET_ROOT" \
     --device "cuda" \
     --window-size 10 \
     --fold-policy default \

graph_net/torch/typical_sequence_split_points.py

@@ -150,12 +150,14 @@ def get_len(tid):
             get_len(sym_id)
         return token2len
 
-    def analyze(self, model_paths_file: str, device: str) -> Dict[str, Dict]:
+    def analyze(
+        self, model_path_prefix: str, model_paths_file: str, device: str
+    ) -> Dict[str, Dict]:
         input_file = Path(model_paths_file)
 
         with open(input_file, "r") as f:
             model_paths = [
-                Path(line.strip())
+                Path(model_path_prefix) / line.strip()
                 for line in f
                 if line.strip() and not line.startswith("#")
             ]
@@ -264,7 +266,7 @@ def main(args):
         fold_policy=args.fold_policy,
         fold_times=args.fold_times,
     )
-    results = analyzer.analyze(args.model_list, args.device)
+    results = analyzer.analyze(args.model_path_prefix, args.model_list, args.device)
     if args.output_json:
         with open(args.output_json, "w") as f:
             json.dump(results, f, indent=4)
@@ -280,6 +282,12 @@ def main(args):
         required=True,
         help="Path to a text file containing paths to models (one per line).",
     )
+    parser.add_argument(
+        "--model-path-prefix",
+        type=str,
+        default="./",
+        help="Prefix to add to each model path in the list.",
+    )
     parser.add_argument(
         "--device",
         type=str,