| |
| """归档: 把 logs/<run> 与权重副本整理到 archive/<date>_<name>/, 生成 summary.txt |
| 用法: python scripts/archive_run.py --name s2_best --src_log logs/s2 --weights weight/s2/net_params_*.pkl |
| """ |
| import argparse, os, shutil, json, glob, datetime |
|
|
| def main(): |
| ap = argparse.ArgumentParser() |
| ap.add_argument("--name", required=True) |
| ap.add_argument("--src_log", default="logs") |
| ap.add_argument("--weights", nargs="*", default=[]) |
| ap.add_argument("--eval_json", default="", help="eval_val.py 输出") |
| args = ap.parse_args() |
| dst = os.path.join("archive", f"{datetime.date.today().isoformat()}_{args.name}") |
| os.makedirs(dst, exist_ok=True) |
| if os.path.isdir(args.src_log): |
| shutil.copytree(args.src_log, os.path.join(dst, "log"), dirs_exist_ok=True) |
| wd = os.path.join(dst, "weights"); os.makedirs(wd, exist_ok=True) |
| for pat in args.weights: |
| for f in glob.glob(pat): |
| shutil.copy2(f, os.path.join(wd, os.path.basename(f))) |
| summary = {"name": args.name, "date": datetime.date.today().isoformat(), "weights": [os.path.basename(f) for pat in args.weights for f in glob.glob(pat)]} |
| if args.eval_json and os.path.exists(args.eval_json): |
| with open(args.eval_json, encoding="utf-8") as fh: |
| summary["eval_mean"] = json.load(fh).get("mean") |
| with open(os.path.join(dst, "summary.txt"), "w", encoding="utf-8") as fh: |
| fh.write(json.dumps(summary, ensure_ascii=False, indent=1)) |
| print("archived to", dst) |
|
|
| if __name__ == "__main__": |
| main()
|
|
|
