소스 검색

accelerate test

justheuristic 3 년 전
부모
커밋
4921097d9d
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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,