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-27 14:45:19 -04:00
parent b77ff3d3a4
commit 0c841cf59a
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

@ -36,7 +36,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* **Automatic training loop**
```python ```python
# define what happens for training here # define what happens for training here
@ -49,7 +49,7 @@ def training_step(self, data_batch, batch_nb):
return {'loss': loss} return {'loss': loss}
``` ```
*Automatic validation loop* **Automatic validation loop**
```python ```python
# define what happens for validation here # define what happens for validation here
@ -62,7 +62,7 @@ def validation_step(self, data_batch, batch_nb):
return {'loss': loss} return {'loss': loss}
``` ```
*Collate the output of the validation_step* **Collate the output of the validation_step**
```python ```python
def validation_end(self, outputs): def validation_end(self, outputs):