Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "ef4e095d247ec55a98ed9ae6e8c4173551b92e26"
Commit 92e6cd93 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 304840194
parent bbdc9810
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/benchmark/bert_squad_benchmark.py
View file @ 92e6cd93
...@@ -128,7 +128,7 @@ class BertSquadBenchmarkBase(benchmark_utils.BertBenchmarkBase): ...@@ -128,7 +128,7 @@ class BertSquadBenchmarkBase(benchmark_utils.BertBenchmarkBase):
eval_metrics = run_squad.eval_squad(strategy=strategy, eval_metrics = run_squad.eval_squad(strategy=strategy,
input_meta_data=input_meta_data) input_meta_data=input_meta_data)
# Use F1 score as reported evaluation metric. # Use F1 score as reported evaluation metric.
self.eval_metrics = eval_metrics['f1'] self.eval_metrics = eval_metrics['final_f1']
class BertSquadBenchmarkReal(BertSquadBenchmarkBase): class BertSquadBenchmarkReal(BertSquadBenchmarkBase):
......
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