Skip to content

Commit f786794

Browse files
committed
Merge branch 'bugfix/6193_dp_dont_reduce_dict' of https://github.com/PyTorchLightning/pytorch-lightning into bugfix/6193_dp_dont_reduce_dict
2 parents 6dbf462 + 6ff54d0 commit f786794

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

tests/accelerators/test_dp.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -130,13 +130,7 @@ def test_dp_training_step_dict(tmpdir):
130130
"""
131131
This test verify dp properly reduce dictionaries
132132
"""
133-
134-
class TestModel(BoringModel):
135-
136-
def training_step(self, batch, batch_idx):
137-
return super().training_step(batch, batch_idx)
138-
139-
model = TestModel()
133+
model = BoringModel()
140134
model.training_step_end = None
141135
trainer = pl.Trainer(
142136
default_root_dir=tmpdir,

0 commit comments

Comments
 (0)