pytorch-lightning
6 строк · 398.0 Байт
1from lightning.app.components.multi_node.base import MultiNode2from lightning.app.components.multi_node.fabric import FabricMultiNode3from lightning.app.components.multi_node.pytorch_spawn import PyTorchSpawnMultiNode4from lightning.app.components.multi_node.trainer import LightningTrainerMultiNode5
6__all__ = ["FabricMultiNode", "MultiNode", "PyTorchSpawnMultiNode", "LightningTrainerMultiNode"]7