diff --git a/crates/ml/examples/train_baseline_supervised.rs b/crates/ml/examples/train_baseline_supervised.rs
index ee8c54c16..91fdd0b6b 100644
--- a/crates/ml/examples/train_baseline_supervised.rs
+++ b/crates/ml/examples/train_baseline_supervised.rs
@@ -565,6 +565,8 @@ fn train_fold(
         metrics::set_epoch_loss(model_name, &fold_str, avg_train_loss);
         metrics::set_validation_loss(model_name, &fold_str, val_loss);
         metrics::set_iteration_seconds(model_name, &fold_str, elapsed.as_secs_f64());
+        metrics::set_learning_rate(model_name, &fold_str, adapter.get_learning_rate());
+        metrics::set_epoch_duration(model_name, &fold_str, elapsed.as_secs_f64());
 
         info!(
             "  Fold {} Epoch {}/{}: train_loss={:.6}, val_loss={:.6}, lr={:.2e}, time={:.1}s",