diff --git a/README.md b/README.md index e4d3a2afcd..487150718c 100644 --- a/README.md +++ b/README.md @@ -128,7 +128,7 @@ from torchvision import transforms import pytorch_lightning as pl ``` -#### Step 1: Define a LightningModule +#### Step 1: Define a LightningModule (nn.Module subclass) A LightningModule defines a full *system* (ie: a GAN, autoencoder, BERT or a simple Image Classifier). ```python