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

Fix broken code in docs (#5859) 

Co-authored-by: Roman Tezikov <roman.tezikov@lamoda.ru>
This commit is contained in:
Roman Tezikov 2021-02-08 19:52:12 +03:00 committed by GitHub
parent bd920b4102
commit 2008d77f28
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
docs/source/extensions/datamodules.rst
View File

@ -61,8 +61,8 @@ Here's a simple PyTorch example:
.. code-block:: python
# regular PyTorch
test_data = MNIST(PATH, train=False, download=True)
train_data = MNIST(PATH, train=True, download=True)
test_data = MNIST(my_path, train=False, download=True)
train_data = MNIST(my_path, train=True, download=True)
train_data, val_data = random_split(train_data, [55000, 5000])
train_loader = DataLoader(train_data, batch_size=32)
@ -75,8 +75,9 @@ The equivalent DataModule just organizes the same exact code, but makes it reusa
class MNISTDataModule(pl.LightningDataModule):
def __init__(self, data_dir: str = PATH, batch_size):
def __init__(self, data_dir: str = "path/to/dir", batch_size: int = 32):
super().__init__()
self.data_dir = data_dir
self.batch_size = batch_size
def setup(self, stage=None):
@ -99,7 +100,7 @@ colleagues or use in different projects.
.. code-block:: python
mnist = MNISTDataModule(PATH)
mnist = MNISTDataModule(my_path)
model = LitClassifier()
trainer = Trainer()