diff --git a/docs/LightningModule/RequiredTrainerInterface.md b/docs/LightningModule/RequiredTrainerInterface.md index abc015caf0..42684176cd 100644 --- a/docs/LightningModule/RequiredTrainerInterface.md +++ b/docs/LightningModule/RequiredTrainerInterface.md @@ -80,6 +80,15 @@ class CoolModel(ptl.LightningModule): def test_dataloader(self): return DataLoader(MNIST(os.getcwd(), train=True, download=True, transform=transforms.ToTensor()), batch_size=32) ``` +--- +### How do these methods fit into the broader training? +The LightningModule interface is on the right. Each method corresponds to a part of a research project. Lightning automates everything not in blue. + +
+ + + +
---