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 4b22284 commit 97dabf8Copy full SHA for 97dabf8
src/lightning/pytorch/strategies/ddp.py
@@ -420,6 +420,11 @@ def teardown(self) -> None:
420
421
422
class MultiModelDDPStrategy(DDPStrategy):
423
+ """Specific strategy for training on multiple models with multiple optimizers (e.g. GAN training).
424
+
425
+ This strategy allows to wrap multiple models with DDP, rather than just one which is about just normal DDPStrategy.
426
+ """
427
428
@override
429
def _setup_model(self, model: Module) -> DistributedDataParallel:
430
device_ids = self.determine_ddp_device_ids()
0 commit comments