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

added docs page

This commit is contained in:
William Falcon 2019-06-26 20:00:53 -04:00
parent 8a0bd25d30
commit 08b95ec6a8
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
README.md
View File

@ -38,13 +38,20 @@ To use lightning do 2 things:
- Automatic training loop - Automatic training loop
```python ```python
# define what happens for training here # define what happens for training here
def training_step(self, data_batch, batch_nb): def training_step(self, data_batch, batch_nb):
x, y = data_batch
out = self.forward(x)
loss = my_loss(out, y)
return {'loss': loss}
``` ```
- Automatic validation loop - Automatic validation loop
```python ```python
# define what happens for validation here # define what happens for validation here
def validation_step(self, data_batch, batch_nb): def validation_step(self, data_batch, batch_nb): x, y = data_batch
out = self.forward(x)
loss = my_loss(out, y)
return {'loss': loss}
``` ```
- Automatic early stopping - Automatic early stopping