From 80759cff3708f7206197243abd7d4f16d78fd88c Mon Sep 17 00:00:00 2001 From: bingshuo <565183519@qq.com> Date: Mon, 18 May 2026 21:16:02 +0800 Subject: [PATCH] watch_distill.py: filter tqdm progress bars, fix step parsing --- scripts/watch_distill.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/scripts/watch_distill.py b/scripts/watch_distill.py index ef891ef..580027d 100644 --- a/scripts/watch_distill.py +++ b/scripts/watch_distill.py @@ -45,6 +45,9 @@ def parse_loss(line): if m: return float(m.group(1)) m = re.search(r"loss[=:]\s*([\d.]+)", line) + if m: + return float(m.group(1)) + m = re.search(r"loss=([\d.]+)", line) if m: return float(m.group(1)) return None @@ -54,6 +57,9 @@ def parse_step(line): m = re.search(r"(\d+)/(\d+)\s+\[", line) if m: return int(m.group(1)), int(m.group(2)) + m = re.search(r"step=(\d+)", line) + if m: + return int(m.group(1)), None return None, None def parse_progress(line): @@ -173,4 +179,4 @@ def main(): sys.exit(0) if __name__ == "__main__": - main() \ No newline at end of file + main()