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

Fix typo (#2907) 

Variable defined as `mnist_dm` but used as `mnist`. Change to use `mnist_dm`.
This commit is contained in:
Shiv Dhar 2020-08-11 12:09:16 +05:30 committed by GitHub
parent 35a3fd2f97
commit 0097630a95
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/introduction_guide.rst
View File

@ -291,7 +291,7 @@ Using DataModules allows easier sharing of full dataset definitions.
# use an MNIST dataset
mnist_dm = MNISTDatamodule()
model = LitModel(num_classes=mnist.num_classes)
model = LitModel(num_classes=mnist_dm.num_classes)
trainer.fit(model, mnist_dm)
# or other datasets with the same model