Răsfoiți Sursa

accelerate test

justheuristic 3 ani în urmă
părinte
comite
4921097d9d
1 a modificat fișierele cu 2 adăugiri și 2 ștergeri
  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,