Log training metrics as MetricEvents through the event logger

miles/utils/tracking_utils/__init__.py

@@ -1,5 +1,10 @@
 import logging
 
+import torch
+
+from miles.utils.event_logger.logger import get_event_logger, is_event_logger_initialized
+from miles.utils.event_logger.models import MetricEvent
+
 from .base import TrackingManager
 
 logger = logging.getLogger(__name__)
@@ -14,6 +19,10 @@ def log(args, metrics, step_key: str):
     step = metrics.get(step_key)
     _manager.log(metrics, step=step, step_key=step_key)
 
+    if is_event_logger_initialized():
+        serializable_metrics = {k: (v.item() if isinstance(v, torch.Tensor) else v) for k, v in metrics.items()}
+        get_event_logger().log(MetricEvent, {"metrics": serializable_metrics}, print_log=False)
+
 
 def finish_tracking():
     _manager.finish()