Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "438c194ec61b34b377727c97d5922897546c581d"
Commit 22f10c60 authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

Add model metrics to the training summary. 

PiperOrigin-RevId: 307914609
parent 2d05fc8a
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/nlp/bert/model_training_utils.py
View file @ 22f10c60
...@@ -491,6 +491,8 @@ def run_customized_training_loop( ...@@ -491,6 +491,8 @@ def run_customized_training_loop(
'total_training_steps': total_training_steps, 'total_training_steps': total_training_steps,
'train_loss': _float_metric_value(train_loss_metric), 'train_loss': _float_metric_value(train_loss_metric),
} }
for metric in model.metrics:
training_summary[metric.name] = _float_metric_value(metric)
if eval_metrics: if eval_metrics:
# TODO(hongkuny): Cleans up summary reporting in text. # TODO(hongkuny): Cleans up summary reporting in text.
training_summary['last_train_metrics'] = _float_metric_value( training_summary['last_train_metrics'] = _float_metric_value(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment