We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 19e055c commit 088dbbeCopy full SHA for 088dbbe
tests/callbacks/test_finetuning_scheduler_callback.py
@@ -412,10 +412,10 @@ def test_finetuningscheduling_misconfiguration(tmpdir, callbacks: List[Callback]
412
@pytest.mark.parametrize(
413
"strategy, gpus, plugins",
414
[
415
- pytest.param("ddp2", 2, None, marks=RunIf(min_gpus=2)),
416
- pytest.param("ddp_fully_sharded", 2, None, marks=RunIf(min_gpus=2)),
417
- pytest.param("horovod", None, None, marks=RunIf(min_gpus=2)),
418
- pytest.param("ddp", 2, "deepspeed_stage_2", marks=RunIf(deepspeed=True, min_gpus=2)),
+ pytest.param("ddp2", 1, None, marks=RunIf(min_gpus=1)),
+ pytest.param("ddp_fully_sharded", 1, None, marks=RunIf(min_gpus=1)),
+ pytest.param("horovod", None, None, marks=RunIf(min_gpus=1)),
+ pytest.param("deepspeed_stage_2", 1, None, marks=RunIf(deepspeed=True, min_gpus=1)),
419
],
420
)
421
def test_finetuningscheduling_distributed_compat(tmpdir, strategy, gpus, plugins):
0 commit comments