Internal change

PiperOrigin-RevId: 304840194

Internal change
PiperOrigin-RevId: 304840194
92e6cd93 · Hongkun Yu · A. Unique TensorFlower · bbdc9810 · 92e6cd93
Commit 92e6cd93 authored Apr 04, 2020 by Hongkun Yu Committed by A. Unique TensorFlower Apr 04, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/benchmark/bert_squad_benchmark.py official/benchmark/bert_squad_benchmark.py +1 -1

No files found.
--- a/official/benchmark/bert_squad_benchmark.py
+++ b/official/benchmark/bert_squad_benchmark.py
@@ -128,7 +128,7 @@ class BertSquadBenchmarkBase(benchmark_utils.BertBenchmarkBase):
    eval_metrics = run_squad.eval_squad(strategy=strategy,
                                        input_meta_data=input_meta_data)
    # Use F1 score as reported evaluation metric.
-    self.eval_metrics = eval_metrics['f1']
+    self.eval_metrics = eval_metrics['final_f1']


 class BertSquadBenchmarkReal(BertSquadBenchmarkBase):