From 37445d11980eab3f310c5e491e671a48d6c6d95b Mon Sep 17 00:00:00 2001
From: Sherlock <baihan.huang@gmail.com>
Date: Wed, 30 Sep 2020 14:30:20 -0700
Subject: [PATCH] Update Bert Perf Script (#5339)

Co-authored-by: Sherlock Huang <bahuang@OrtTrainingDev3.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
---
 orttraining/tools/ci_test/run_bert_perf_test.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/orttraining/tools/ci_test/run_bert_perf_test.py b/orttraining/tools/ci_test/run_bert_perf_test.py
index d3909f36f5..170f952384 100644
--- a/orttraining/tools/ci_test/run_bert_perf_test.py
+++ b/orttraining/tools/ci_test/run_bert_perf_test.py
@@ -26,10 +26,10 @@ def main():
 
     Config = namedtuple('Config', ['use_mixed_precision', 'max_seq_length', 'batch_size', 'max_predictions_per_seq'])
     configs = [
-        Config(True, 128, 64, 20),
-        Config(True, 512, 10, 80),
-        Config(False, 128, 33, 20),
-        Config(False, 512, 5, 80)
+        Config(True, 128, 76, 20),
+        Config(True, 512, 11, 80),
+        Config(False, 128, 39, 20),
+        Config(False, 512, 6, 80)
     ]
 
     # run BERT training