saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update Bert Perf Script (#5339)

Browse source 
Co-authored-by: Sherlock Huang <bahuang@OrtTrainingDev3.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
This commit is contained in:
Sherlock 2020-09-30 14:30:20 -07:00 committed by GitHub
parent 8d4740b39c
commit 37445d1198
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
orttraining/tools/ci_test/run_bert_perf_test.py
View file

@ -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