Explicit data prep calls in datamodule docs (#698)

Lightning-Universe · Jul 28, 2021 · 11686f2 · 11686f2
1 parent e22b8d0
commit 11686f2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/docs/source/datamodules.rst b/docs/source/datamodules.rst
@@ -26,6 +26,9 @@ Or use it manually with plain PyTorch
 Example::
 
     dm = MNISTDataModule('path/to/data')
+    # download data and setup dataloaders
+    dm.prepare_data()
+    dm.setup()
     for batch in dm.train_dataloader():
         ...
     for batch in dm.val_dataloader():