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

Fix imports for lightning cli examples (#16871)

This commit is contained in:
Darren Tuit 2023-02-26 12:09:34 +13:00 committed by GitHub
parent df6e37da1c
commit cdf21a1305
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-pytorch/cli/lightning_cli_intermediate_2.rst
View File

@ -57,7 +57,7 @@ To support multiple models, when instantiating ``LightningCLI`` omit the ``model
# main.py # main.py
from pytorch_lightning.cli import LightningCLI from pytorch_lightning.cli import LightningCLI
from pytorch_lightning.demos.boring_classes import DemoModel from pytorch_lightning.demos.boring_classes import DemoModel, BoringDataModule
class Model1(DemoModel): class Model1(DemoModel):
@ -101,7 +101,7 @@ To support multiple data modules, when instantiating ``LightningCLI`` omit the `
# main.py # main.py
import torch import torch
from pytorch_lightning.cli import LightningCLI from pytorch_lightning.cli import LightningCLI
from pytorch_lightning.demos.boring_classes import BoringDataModule from pytorch_lightning.demos.boring_classes import DemoModel, BoringDataModule
class FakeDataset1(BoringDataModule): class FakeDataset1(BoringDataModule):