We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 03ebd18 commit 73f2a2bCopy full SHA for 73f2a2b
project/lit_image_classifier.py
@@ -7,7 +7,7 @@
7
import torch.nn as nn
8
from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint, LearningRateMonitor
9
from pytorch_lightning.metrics import Accuracy
10
-from pytorch_lightning.plugins import DDPPlugin
+from pytorch_lightning.plugins.ddp_plugin import DDPPlugin
11
from torch.optim.lr_scheduler import CosineAnnealingLR
12
from torch.utils.data import DataLoader, random_split
13
from torchvision import transforms
0 commit comments