From ddacfbbff1593f984fd77f6c374aca0e4403fc82 Mon Sep 17 00:00:00 2001 From: jgrusewski Date: Tue, 3 Mar 2026 00:10:00 +0100 Subject: [PATCH] feat(supervised): record learning rate and epoch duration metrics Co-Authored-By: Claude Opus 4.6 --- crates/ml/examples/train_baseline_supervised.rs | 2 ++ 1 file changed, 2 insertions(+) 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",