Improve test speeds by up to 30% in multi-gpu settings (#2830)

huggingface · Jun 6, 2024 · 05d240a · 05d240a
1 parent bad2ce4
commit 05d240a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):