From 456f85546a53993b00def6d1d3c25a20c31bf604 Mon Sep 17 00:00:00 2001 From: Juhan Bae Date: Wed, 20 Mar 2024 13:06:26 -0400 Subject: [PATCH] Larger BS --- examples/glue/analyze.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/glue/analyze.py b/examples/glue/analyze.py index a52eb7b..472d724 100644 --- a/examples/glue/analyze.py +++ b/examples/glue/analyze.py @@ -151,7 +151,7 @@ def main(): per_device_batch_size=None, factor_args=factor_args, overwrite_output_dir=True, - initial_per_device_batch_size_attempt=64, + initial_per_device_batch_size_attempt=128, ) analyzer.compute_pairwise_scores( scores_name="pairwise",