We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7d1ca37 commit ffe7504Copy full SHA for ffe7504
tests/tests_fabric/strategies/test_ddp.py
@@ -16,6 +16,7 @@
16
17
import pytest
18
import torch
19
+from torch.nn.parallel import DistributedDataParallel
20
21
from lightning_fabric.strategies import DDPStrategy
22
from lightning_fabric.strategies.ddp import _DDPBackwardSyncControl
0 commit comments