diff --git a/example/run_all.py b/example/run_all.py new file mode 100644 index 0000000..7926852 --- /dev/null +++ b/example/run_all.py @@ -0,0 +1,81 @@ +#!/usr/bin/env python3 +"""One-command pipeline runner with config-driven paths.""" + +import argparse +import json +import subprocess +import sys +from pathlib import Path + +from platform_utils import safe_path, is_windows, resolve_path + + +def run(cmd): + print("running:", " ".join(cmd)) + cmd = [safe_path(arg) for arg in cmd] + if is_windows(): + subprocess.run(cmd, check=True, shell=False) + else: + subprocess.run(cmd, check=True) + + +def parse_args(): + parser = argparse.ArgumentParser(description="Run prepare -> train -> export -> evaluate.") + base_dir = Path(__file__).resolve().parent + parser.add_argument("--config", default=str(base_dir / "config.json")) + parser.add_argument("--device", default="auto", help="cpu, cuda, or auto") + parser.add_argument("--skip-prepare", action="store_true") + parser.add_argument("--skip-train", action="store_true") + parser.add_argument("--skip-export", action="store_true") + parser.add_argument("--skip-eval", action="store_true") + return parser.parse_args() + + +def main(): + args = parse_args() + base_dir = Path(__file__).resolve().parent + config_path = Path(args.config) + with open(config_path, "r", encoding="utf-8") as f: + cfg = json.load(f) + + config_path = resolve_path(base_dir, config_path) + timesteps = cfg.get("timesteps", 200) + seq_len = cfg.get("sample_seq_len", cfg.get("seq_len", 64)) + batch_size = cfg.get("sample_batch_size", cfg.get("batch_size", 2)) + clip_k = cfg.get("clip_k", 5.0) + + if not args.skip_prepare: + run([sys.executable, str(base_dir / "prepare_data.py")]) + if not args.skip_train: + run([sys.executable, str(base_dir / "train.py"), "--config", str(config_path), "--device", args.device]) + if not args.skip_export: + run( + [ + sys.executable, + str(base_dir / "export_samples.py"), + "--include-time", + "--device", + args.device, + "--config", + str(config_path), + "--timesteps", + str(timesteps), + "--seq-len", + str(seq_len), + "--batch-size", + str(batch_size), + "--clip-k", + str(clip_k), + "--use-ema", + ] + ) + if not args.skip_eval: + ref = cfg.get("data_glob") or cfg.get("data_path") or "" + if ref: + run([sys.executable, str(base_dir / "evaluate_generated.py"), "--reference", str(ref)]) + else: + run([sys.executable, str(base_dir / "evaluate_generated.py")]) + + +if __name__ == "__main__": + main()