Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed Import in Docs For Multinode Trainer Name Which does Not Exist (#15663)

This commit is contained in:
Rick Izzo 2022-11-12 19:17:29 -05:00 committed by GitHub
parent 57ac5483a0
commit 23f88cd0a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source-app/levels/basic/hello_components/pl_multinode.py
View File

@ -1,6 +1,6 @@
# app.py
import lightning as L
from lightning.app.components import PyTorchLightningMultiNode
from lightning.app.components import LightningTrainerMultiNode
from lightning.pytorch.demos.boring_classes import BoringModel
@ -12,7 +12,7 @@ class LightningTrainerDistributed(L.LightningWork):
trainer.fit(model)
# 8 GPU: (2 nodes of 4 x v100)
component = PyTorchLightningMultiNode(
component = LightningTrainerMultiNode(
LightningTrainerDistributed,
num_nodes=2,
cloud_compute=L.CloudCompute("gpu-fast-multi"), # 4 x v100