wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

added docs page

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+9 -2
View File
@@ -38,13 +38,20 @@ To use lightning do 2 things:
- Automatic training loop
```python
# 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
```python
# 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