From b92c09052aaa32136a100e015379158ded7ec175 Mon Sep 17 00:00:00 2001 From: Nikil Ravi Date: Thu, 9 Jan 2025 04:40:32 -0800 Subject: [PATCH] wip --- src/levanter/eval_harness.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/levanter/eval_harness.py b/src/levanter/eval_harness.py index b006ed5db..f6ed75f9e 100644 --- a/src/levanter/eval_harness.py +++ b/src/levanter/eval_harness.py @@ -483,6 +483,12 @@ def _compute_averages(outputs): Returns: Averages dictionary with macro and micro averages for all metrics. """ + + print("Outputs: ", outputs) + logger.info("Computing averages...") + logger.info("Outputs: ", outputs) + logger.info("Results: ", outputs["results"]) + averages = {} metric_keys = set()