Update README.md
This commit is contained in:
parent
c0e26b8766
commit
2db86381c6
|
@ -128,7 +128,7 @@ from torchvision import transforms
|
||||||
import pytorch_lightning as pl
|
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).
|
A LightningModule defines a full *system* (ie: a GAN, autoencoder, BERT or a simple Image Classifier).
|
||||||
|
|
||||||
```python
|
```python
|
||||||
|
|
Loading…
Reference in New Issue