12 lines
338 B
Python
12 lines
338 B
Python
|
from lightning import LightningApp
|
||
|
|
from lightning.app.components.training import LightningTrainingComponent
|
||
|
|
from lightning.app.utilities.packaging.cloud_compute import CloudCompute
|
||
|
|
|
||
|
|
app = LightningApp(
|
||
|
|
LightningTrainingComponent(
|
||
|
|
"train.py",
|
||
|
|
num_nodes=2,
|
||
|
|
cloud_compute=CloudCompute("gpu-fast-multi"),
|
||
|
|
),
|
||
|
|
)
|