From 05d240af95f58f636a4c3798e7fe51fb8943163e Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Thu, 6 Jun 2024 06:12:59 -0400 Subject: [PATCH] Improve test speeds by up to 30% in multi-gpu settings (#2830) --- src/accelerate/test_utils/testing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/accelerate/test_utils/testing.py b/src/accelerate/test_utils/testing.py index 777ea83662e..41e3ac35115 100644 --- a/src/accelerate/test_utils/testing.py +++ b/src/accelerate/test_utils/testing.py @@ -99,7 +99,7 @@ def get_launch_command(**kwargs) -> list: return command -DEFAULT_LAUNCH_COMMAND = get_launch_command(num_processes=device_count) +DEFAULT_LAUNCH_COMMAND = get_launch_command(num_processes=device_count, monitor_interval=0.1) def parse_flag_from_env(key, default=False):