bingshuo/guanghulab
1
0
Fork 0
Code Issues 2 Pull Requests Actions 1 Packages Projects Releases Wiki Activity

watch_distill.py: filter tqdm progress bars, fix step parsing

Browse Source
This commit is contained in:
bingshuo 2026-05-18 21:16:02 +08:00
parent 7d53003fb1
commit 80759cff37
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/watch_distill.py
View File

@ -45,6 +45,9 @@ def parse_loss(line):
if m: if m:
return float(m.group(1)) return float(m.group(1))
m = re.search(r"loss[=:]\s*([\d.]+)", line) m = re.search(r"loss[=:]\s*([\d.]+)", line)
if m:
return float(m.group(1))
m = re.search(r"loss=([\d.]+)", line)
if m: if m:
return float(m.group(1)) return float(m.group(1))
return None return None
@ -54,6 +57,9 @@ def parse_step(line):
m = re.search(r"(\d+)/(\d+)\s+\[", line) m = re.search(r"(\d+)/(\d+)\s+\[", line)
if m: if m:
return int(m.group(1)), int(m.group(2)) 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 return None, None
def parse_progress(line): def parse_progress(line):
@ -173,4 +179,4 @@ def main():
sys.exit(0) sys.exit(0)
if __name__ == "__main__": if __name__ == "__main__":
main() main()