From 012aad3167aaccd974886b8613ce501ccb9eedec Mon Sep 17 00:00:00 2001 From: "ferdinand.mom" Date: Sun, 1 Dec 2024 03:42:14 +0000 Subject: [PATCH] fix extract metrics --- extract_metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extract_metrics.py b/extract_metrics.py index b26b08e..90dbbce 100644 --- a/extract_metrics.py +++ b/extract_metrics.py @@ -63,7 +63,7 @@ def process_file(filepath): tokens_s_gpu_values = [] with open(filepath, 'r') as f: for line in f: - if '[default0]:[rank 0]' in line: + if re.search(r'\[default\d+\]:\[rank \d+\]', line): tokens_s_gpu = parse_log_line(line) if tokens_s_gpu is not None: tokens_s_gpu_values.append(tokens_s_gpu)