From 2db86381c6e9468828418cc77ef203486ae8f72f Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 23 Sep 2020 07:39:46 -0400 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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