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 19:48:50 -04:00
parent 92f5e026bb
commit 71af821b85
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -41,8 +41,10 @@ To use lightning do 2 things:
from pytorch_lightning import Trainer from pytorch_lightning import Trainer
from pytorch_lightning.utils.pt_callbacks import EarlyStopping, ModelCheckpoint from pytorch_lightning.utils.pt_callbacks import EarlyStopping, ModelCheckpoint
# 1 - look at the this page for the interface (https://williamfalcon.github.io/pytorch-lightning/)
model = MyModel() model = MyModel()
# 2 - feed to trainer
trainer = Trainer( trainer = Trainer(
checkpoint_callback=ModelCheckpoint(...), checkpoint_callback=ModelCheckpoint(...),
early_stop_callback=EarlyStopping(...), early_stop_callback=EarlyStopping(...),