Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
William Falcon 2019-06-29 18:06:30 -04:00 committed by GitHub
parent 614d84e560
commit a36061ad2b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -38,7 +38,7 @@ To use lightning do 2 things:
## What does lightning control for me?
Everything! Except you define your data and what happens inside the training and validation loop.
**What happens in the training loop**
**Could be as complex as seq-2-seq + attention**
```python
# define what happens for training here
@ -65,7 +65,7 @@ def training_step(self, data_batch, batch_nb):
return {'loss': loss}
```
**What happens in the validation loop**
**Or as basic as CNN image classification**
```python
# define what happens for validation here
@ -78,7 +78,7 @@ def validation_step(self, data_batch, batch_nb):
return {'loss': loss}
```
**And what to do with the output of all validation batches**
**And you also decide how to collate the output of all validation steps**
```python
def validation_end(self, outputs):