gitadmin/foxhunt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(supervised): record learning rate and epoch duration metrics

Browse Source 
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
jgrusewski
2026-03-03 00:10:00 +01:00
parent afa506495f
commit ddacfbbff1
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
crates/ml/examples/train_baseline_supervised.rs
View File

@@ -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",