Remove redundant fit call from accelerator connector test ()

This commit is contained in:
Adrian Wälchli 2021-11-19 07:49:52 +01:00 committed by GitHub
parent 137b62d80d
commit c09c9c7607
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions

View File

@ -350,7 +350,6 @@ def test_accelerator_choice_ddp_cpu_and_strategy_spawn(tmpdir):
def _test_accelerator_choice_ddp_cpu_and_strategy(tmpdir, ddp_strategy_class):
model = BoringModel()
trainer = Trainer(
default_root_dir=tmpdir,
strategy=ddp_strategy_class(find_unused_parameters=True),
@ -362,7 +361,6 @@ def _test_accelerator_choice_ddp_cpu_and_strategy(tmpdir, ddp_strategy_class):
assert isinstance(trainer.accelerator, CPUAccelerator)
assert trainer.training_type_plugin.num_processes == 2
assert trainer.training_type_plugin.parallel_devices == [torch.device("cpu")] * 2
trainer.fit(model)
@mock.patch.dict(