Increase batch size for bert to decrease fluctuations (#236)

Increase batch size for bert to decrease fluctuations
hidet-org · Jul 22, 2024 · 3a6b8f1 · 3a6b8f1
1 parent e1fb8b0
commit 3a6b8f1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/benchmarks/run_configs.json b/tests/benchmarks/run_configs.json
@@ -4,8 +4,8 @@
         "id": 1,
         "name": "bert-base-uncased",
         "runfile": "bench_transformer.py",
-        "param_id": 1,
-        "param_name": "seqlen=256",
+        "param_id": 14,
+        "param_name": "bs=16,seqlen=256",
         "dtype_id": 1,
         "dtype_name": "float16"
     },
@@ -189,4 +189,4 @@
         "dtype_id": 1,
         "dtype_name": "float16"
     }
-]
+]