Fix BERT benchmark test.

PiperOrigin-RevId: 273795511

Fix BERT benchmark test.
PiperOrigin-RevId: 273795511
f5e2211f · A. Unique TensorFlower · 7207422d · f5e2211f
Commit f5e2211f authored Oct 09, 2019 by A. Unique TensorFlower
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/official/benchmark/bert_benchmark.py
+++ b/official/benchmark/bert_benchmark.py
@@ -76,7 +76,7 @@ class BertClassifyBenchmarkBase(benchmark_utils.BertBenchmarkBase):

    steps_per_loop = 1

-    run_classifier.run_customized_training(
+    run_classifier.run_bert_classifier(
        strategy,
        bert_config,
        input_meta_data,