From 44b4dfee401a03c0cf3bebfec8700d9b61eb231f Mon Sep 17 00:00:00 2001 From: Md Fahim Faysal Khan Date: Thu, 5 Sep 2024 21:47:28 -0700 Subject: [PATCH] fix tensorboard events dir path (#1032) Fixed the tensorboard dir path after a recent change in MaxText software: https://github.com/google/maxtext/pull/863 --- .github/workflows/baselines/test_maxtext_metrics.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/baselines/test_maxtext_metrics.py b/.github/workflows/baselines/test_maxtext_metrics.py index bd180ecfe..a130c86c6 100644 --- a/.github/workflows/baselines/test_maxtext_metrics.py +++ b/.github/workflows/baselines/test_maxtext_metrics.py @@ -19,7 +19,7 @@ def test_loss(baseline_filename): baseline_filepath = os.path.join(baselines_dir, baseline_filename) test_config = baseline_filename.split(".")[0] - event_file = os.path.join(results_dir, test_config, "logdir/tensorboard/events*") + event_file = os.path.join(results_dir, test_config, "logdir/tensorboard/logdir/events*") event_file = glob.glob(event_file)[0] with open(baseline_filepath, "r") as baseline_file: end_step = json.load(baseline_file)["end_step"] @@ -31,7 +31,7 @@ def test_loss(baseline_filename): def test_step_time(baseline_filename): baseline_filepath = os.path.join(baselines_dir, baseline_filename) test_config = baseline_filename.split(".")[0] - event_file = os.path.join(results_dir, test_config, "logdir/tensorboard/events*") + event_file = os.path.join(results_dir, test_config, "logdir/tensorboard/logdir/events*") event_file = glob.glob(event_file)[0] with open(baseline_filepath, "r") as baseline_file: step_time_avg_expected = json.load(baseline_file)["step_time_avg"]