From 0810a5edf7b6bd4ac472e8f9e3d878da299905f9 Mon Sep 17 00:00:00 2001 From: monoxgas Date: Fri, 9 May 2025 10:53:59 -0600 Subject: [PATCH] Log trainer callback metrics to run --- dreadnode/integrations/transformers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dreadnode/integrations/transformers.py b/dreadnode/integrations/transformers.py index 2df908c3..861b481a 100644 --- a/dreadnode/integrations/transformers.py +++ b/dreadnode/integrations/transformers.py @@ -123,7 +123,7 @@ def on_epoch_begin( if self._run is None or state.epoch is None: return - dn.log_metric("epoch", state.epoch) + dn.log_metric("epoch", state.epoch, to="run") self._epoch_span = dn.task_span(f"Epoch {state.epoch}") self._epoch_span.__enter__() @@ -149,7 +149,7 @@ def on_step_begin( if self._run is None: return - dn.log_metric("step", state.global_step) + dn.log_metric("step", state.global_step, to="run") self._step_span = dn.span(f"Step {state.global_step}") self._step_span.__enter__() @@ -178,6 +178,6 @@ def on_log( for key, value in _clean_keys(logs).items(): if isinstance(value, float | int): - dn.log_metric(key, value, step=state.global_step) + dn.log_metric(key, value, step=state.global_step, to="run") dn.push_update()