From 0097630a95bddc48d6fb5d3b9a58aef2e8e89b22 Mon Sep 17 00:00:00 2001 From: Shiv Dhar Date: Tue, 11 Aug 2020 12:09:16 +0530 Subject: [PATCH] Fix typo (#2907) Variable defined as `mnist_dm` but used as `mnist`. Change to use `mnist_dm`. --- docs/source/introduction_guide.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/introduction_guide.rst b/docs/source/introduction_guide.rst index cb18929df9..abca6333cb 100644 --- a/docs/source/introduction_guide.rst +++ b/docs/source/introduction_guide.rst @@ -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