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

adjust bs size (#4375)

Browse source 
Co-authored-by: Ethan Tao <ettao@microsoft.com>
This commit is contained in:
ytaous 2020-06-30 10:29:48 -07:00 committed by GitHub
parent 89c6da99b5
commit 4380b8ba68
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -26,7 +26,7 @@ def main():
Config = namedtuple('Config', ['use_mixed_precision', 'max_seq_length', 'batch_size', 'max_predictions_per_seq'])
configs = [
Config(True, 128, 66, 20),
Config(True, 128, 64, 20),
Config(True, 512, 10, 80),
Config(False, 128, 33, 20),
Config(False, 512, 5, 80)