changed read me
This commit is contained in:
parent
2d036b3264
commit
198190b71d
|
@ -38,7 +38,7 @@ To use lightning do 2 things:
|
|||
## What does lightning control for me?
|
||||
Everything! Except the following three things:
|
||||
|
||||
**Automatic training loop**
|
||||
**What happens in the training loop**
|
||||
|
||||
```python
|
||||
# define what happens for training here
|
||||
|
@ -51,7 +51,7 @@ def training_step(self, data_batch, batch_nb):
|
|||
return {'loss': loss}
|
||||
```
|
||||
|
||||
**Automatic validation loop**
|
||||
**What happens in the validation loop**
|
||||
|
||||
```python
|
||||
# define what happens for validation here
|
||||
|
@ -64,7 +64,7 @@ def validation_step(self, data_batch, batch_nb):
|
|||
return {'loss': loss}
|
||||
```
|
||||
|
||||
**Collate the output of the validation_step**
|
||||
**And what to do with the output of all validation batches**
|
||||
|
||||
```python
|
||||
def validation_end(self, outputs):
|
||||
|
|
Loading…
Reference in New Issue