wip

stanford-crfm · Jan 9, 2025 · b92c090 · b92c090
1 parent 60b8e16
commit b92c090
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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()