Browse Source

accelerate test

justheuristic 3 years ago
parent
commit
4921097d9d
1 changed files with 2 additions and 2 deletions
  1. 2 2
      tests/test_optimizer.py

+ 2 - 2
tests/test_optimizer.py

@@ -288,8 +288,8 @@ def test_progress_tracker():
 
 @pytest.mark.forked
 def test_optimizer(
-    num_peers: int = 2,
-    num_clients: int = 1,
+    num_peers: int = 1,
+    num_clients: int = 0,
     target_batch_size: int = 64,
     total_epochs: int = 3,
     reuse_grad_buffers: bool = True,