Shortcuts

Lightning Trainer

The Lightning Trainer offers many advanced features and training optimization without changing your models.

Any Trainer flag can be used in Lightning transformers by passing trainer.FLAG=VALUE to train.py.

Max epochs

python train.py task=nlp/text_classification dataset=nlp/text_classification/emotion trainer.max_epochs=4

16-bit precision

python train.py task=nlp/text_classification dataset=nlp/text_classification/emotion trainer.precision=16

Multi GPU

python train.py task=nlp/translation dataset=nlp/translation/wmt16 trainer.gpus=4

TPUs

python train.py task=nlp/translation dataset=nlp/translation/wmt16 trainer.tpu_cores=8

See the Pytorch Lightning Trainer or conf/trainer/default for all the Lightning Trainer options.

Read the Docs v: stable
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.