Skip to content

Commit 73f2a2b

Browse files
committed
Fix import for DDPPlugin
1 parent 03ebd18 commit 73f2a2b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

project/lit_image_classifier.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import torch.nn as nn
88
from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint, LearningRateMonitor
99
from pytorch_lightning.metrics import Accuracy
10-
from pytorch_lightning.plugins import DDPPlugin
10+
from pytorch_lightning.plugins.ddp_plugin import DDPPlugin
1111
from torch.optim.lr_scheduler import CosineAnnealingLR
1212
from torch.utils.data import DataLoader, random_split
1313
from torchvision import transforms

0 commit comments

Comments
 (0)