fix: use correct key best_val_acc in training completion message

This commit is contained in:
KoalaTrapLord authored and GitHub committed 2026-03-21 15:21:35 -04:00
1 parent 06f007e605
commit 6612f3d213
1 file changed
+2 -2
+2 -2
View File
@@ -362,8 +362,8 @@ def _run_training_pipeline(
"checkpoint": summary.get("checkpoint_path", ""),
"device": device,
}
_training_state.message = f"Training complete! Best val acc: {summary.get('best_val_accuracy', 0):.1%}"
log_ml.info(f"Complete: {summary.get('best_val_accuracy', 0):.1%} val accuracy")
_training_state.message = f"Training complete! Best val acc: {summary.get('best_val_acc', 0):.1%}"
log_ml.info(f"Complete: {summary.get('best_val_acc', 0):.1%} val accuracy")
# Auto-reload policy server
if _policy_service and _policy_service._loaded:
_policy_service.reload(summary.get("checkpoint_path"))