Skip to content

Commit 1769e28

Browse files
ethanwharrisSeanNaren
authored andcommitted
Fix DDP_SPAWN compatibility with bug_report_model.py (#6892)
(cherry picked from commit 87f0aea)
1 parent 9b6374f commit 1769e28

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

pl_examples/bug_report_model.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -126,12 +126,13 @@ def configure_optimizers(self):
126126
# args = parser.parse_args(opt)
127127

128128

129-
def test_run():
129+
class TestModel(BoringModel):
130+
131+
def on_train_epoch_start(self) -> None:
132+
print('override any method to prove your bug')
130133

131-
class TestModel(BoringModel):
132134

133-
def on_train_epoch_start(self) -> None:
134-
print('override any method to prove your bug')
135+
def test_run():
135136

136137
# fake data
137138
train_data = torch.utils.data.DataLoader(RandomDataset(32, 64))

0 commit comments

Comments
 (0)