Skip to content

GitLab

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

Fix BERT benchmark test. 

PiperOrigin-RevId: 273795511
parent 7207422d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/benchmark/bert_benchmark.py
View file @ f5e2211f
...@@ -76,7 +76,7 @@ class BertClassifyBenchmarkBase(benchmark_utils.BertBenchmarkBase): ...@@ -76,7 +76,7 @@ class BertClassifyBenchmarkBase(benchmark_utils.BertBenchmarkBase):
steps_per_loop = 1 steps_per_loop = 1
run_classifier.run_customized_training( run_classifier.run_bert_classifier(
strategy, strategy,
bert_config, bert_config,
input_meta_data, input_meta_data,
......
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