Skip to content

Commit 8baf30e

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent c3b3a78 commit 8baf30e

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

examples/pytorch/bug_report/bug_report_model.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,8 @@
1-
import os
21

32
import torch
4-
from lightning.pytorch import LightningModule, Trainer
5-
from torch.utils.data import DataLoader, Dataset
6-
3+
from lightning.pytorch import LightningModule
74
from pytorch.cli import LightningCLI
5+
from torch.utils.data import DataLoader, Dataset
86

97

108
class RandomDataset(Dataset):
@@ -49,6 +47,7 @@ def run():
4947
val_data = DataLoader(RandomDataset(32, 64), batch_size=2)
5048

5149
import os
50+
5251
os.environ["PL_TRAINER_ACCELERATOR"] = "cpu"
5352
os.environ["PL_TRAINER_DEVICES"] = "2"
5453

0 commit comments

Comments
 (0)