diff --git a/README.md b/README.md index cb46703293..36bf792865 100644 --- a/README.md +++ b/README.md @@ -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):