Skip to content

GitLab

Commit 4dbdb450 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 267607964
parent 45261f52
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/bert/model_training_utils.py
View file @ 4dbdb450
...@@ -413,7 +413,7 @@ def run_customized_training_loop( ...@@ -413,7 +413,7 @@ def run_customized_training_loop(
# 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(
train_metrics[0]) train_metrics[0])
training_summary['eval_metricss'] = _float_metric_value(eval_metrics[0]) training_summary['eval_metrics'] = _float_metric_value(eval_metrics[0])
_write_txt_summary(training_summary, model_dir) _write_txt_summary(training_summary, model_dir)
......
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