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

changed read me

This commit is contained in:
William Falcon 2019-06-28 14:14:21 -04:00
parent 2d036b3264
commit 198190b71d
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? ## What does lightning control for me?
Everything! Except the following three things: Everything! Except the following three things:
**Automatic training loop** **What happens in the training loop**
```python ```python
# define what happens for training here # define what happens for training here
@ -51,7 +51,7 @@ def training_step(self, data_batch, batch_nb):
return {'loss': loss} return {'loss': loss}
``` ```
**Automatic validation loop** **What happens in the validation loop**
```python ```python
# define what happens for validation here # define what happens for validation here
@ -64,7 +64,7 @@ def validation_step(self, data_batch, batch_nb):
return {'loss': loss} return {'loss': loss}
``` ```
**Collate the output of the validation_step** **And what to do with the output of all validation batches**
```python ```python
def validation_end(self, outputs): def validation_end(self, outputs):