diff --git a/opacus/tests/accountants_test.py b/opacus/tests/accountants_test.py index 82139c2c..1f8fafee 100644 --- a/opacus/tests/accountants_test.py +++ b/opacus/tests/accountants_test.py @@ -136,7 +136,7 @@ def test_get_noise_multiplier_prv_steps(self) -> None: ), delta=st.sampled_from([1e-4, 1e-5, 1e-6]), ) - @settings(deadline=10000) + @settings(deadline=40000) def test_get_noise_multiplier_overshoot(self, epsilon, epochs, sample_rate, delta): noise_multiplier = get_noise_multiplier( target_epsilon=epsilon, diff --git a/opacus/tests/batch_memory_manager_test.py b/opacus/tests/batch_memory_manager_test.py index 563c3795..b4d73767 100644 --- a/opacus/tests/batch_memory_manager_test.py +++ b/opacus/tests/batch_memory_manager_test.py @@ -59,7 +59,7 @@ def _init_training(self, batch_size=10, **data_loader_kwargs): batch_size=st.sampled_from([8, 16, 64]), max_physical_batch_size=st.sampled_from([4, 8]), ) - @settings(suppress_health_check=list(HealthCheck), deadline=10000) + @settings(suppress_health_check=list(HealthCheck), deadline=40000) def test_basic( self, num_workers: int, @@ -119,7 +119,7 @@ def test_basic( num_workers=st.integers(0, 4), pin_memory=st.booleans(), ) - @settings(suppress_health_check=list(HealthCheck), deadline=10000) + @settings(suppress_health_check=list(HealthCheck), deadline=40000) def test_empty_batch( self, num_workers: int, diff --git a/opacus/tests/per_sample_gradients_utils_test.py b/opacus/tests/per_sample_gradients_utils_test.py index ef5b4ce7..a5e63aa9 100644 --- a/opacus/tests/per_sample_gradients_utils_test.py +++ b/opacus/tests/per_sample_gradients_utils_test.py @@ -71,7 +71,7 @@ def per_sample_grads_utils_test( groups=st.integers(1, 12), grad_sample_mode=st.sampled_from(get_grad_sample_modes(use_ew=True)), ) - @settings(deadline=10000) + @settings(deadline=40000) def test_conv1d( self, N: int, @@ -120,7 +120,7 @@ def test_conv1d( batch_first=st.booleans(), grad_sample_mode=st.sampled_from(get_grad_sample_modes(use_ew=True)), ) - @settings(deadline=10000) + @settings(deadline=40000) def test_linear( self, N: int,