Add model metrics to the training summary.

PiperOrigin-RevId: 307914609

Add model metrics to the training summary.
PiperOrigin-RevId: 307914609
22f10c60 · Chen Chen · A. Unique TensorFlower · 2d05fc8a · 22f10c60
Commit 22f10c60 authored Apr 22, 2020 by Chen Chen Committed by A. Unique TensorFlower Apr 22, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

official/nlp/bert/model_training_utils.py official/nlp/bert/model_training_utils.py +2 -0

No files found.
--- a/official/nlp/bert/model_training_utils.py
+++ b/official/nlp/bert/model_training_utils.py
@@ -491,6 +491,8 @@ def run_customized_training_loop(
        'total_training_steps': total_training_steps,
        'train_loss': _float_metric_value(train_loss_metric),
    }
+    for metric in model.metrics:
+      training_summary[metric.name] = _float_metric_value(metric)
    if eval_metrics:
      # TODO(hongkuny): Cleans up summary reporting in text.
      training_summary['last_train_metrics'] = _float_metric_value(