mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-12 12:40:20 +08:00
changed read me
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user