Quellcode durchsuchen

accelerate test

justheuristic vor 3 Jahren
Ursprung
Commit
4921097d9d
1 geänderte Dateien mit 2 neuen und 2 gelöschten Zeilen
  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,