Increase batch size

pomonam · Mar 20, 2024 · 0b6811b · 0b6811b
1 parent 456f855
commit 0b6811b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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=128,
+        initial_per_device_batch_size_attempt=256,
     )
     analyzer.compute_pairwise_scores(
         scores_name="pairwise",