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

updated docs

Browse Source
This commit is contained in:
William Falcon
2019-08-15 21:29:25 -04:00
parent 90f01c05bc
commit 4b97319c2e
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/index.md
+6 -5
View File
@@ -1,7 +1,6 @@
###### New project Quick Start ###### New project Quick Start
To start a new project you define two files, a LightningModule and a Trainer file. To start a new project define two files, a LightningModule and a Trainer file.
To illustrate Lightning power and simplicity, here's an example of a typical research flow.
Here's an example of how you would use Lightning for research.
###### Case 1: BERT ###### Case 1: BERT
Let's say you're working on something like BERT but want to try different ways of training or even different networks. Let's say you're working on something like BERT but want to try different ways of training or even different networks.
@@ -52,8 +51,10 @@ trainer = Trainer(gpus=[0, 1, 2, 3], use_amp=True)
trainer.fit(model) trainer.fit(model)
``` ```
Notice that without writing any GPU or 16-bit specific code, your models gain that capability by Notice a few things about this flow:
using Lightning. 1. You're writing pure PyTorch... no unnecessary abstractions or new libraries to learn.
2. You get free GPU and 16-bit support without writing any of that code in your model.
3. You also get all of the capabilities below (without coding or testing yourself).
--- ---
###### Templates ###### Templates