mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-08-21 10:29:22 +08:00

Files

T

History

Jirka Borovec ea59a99426 update org paths & convert logos (#685 )

* fix typos

* update org paths

* update links from READMe to docs

* add svg logo

* add svg logo-text

* update logos

* testing temp paths

* prune links from readme

* optimize imports

* update logo

* update paths in README

* missing imports

2020-01-20 14:50:31 -05:00

__init__.py

changes examples to pl_examples for name connflict

2019-10-19 00:41:17 +02:00

cpu_template.py

Minor imports cleaning (#402 )

2019-10-22 11:32:40 +03:00

gpu_template.py

Minor imports cleaning (#402 )

2019-10-22 11:32:40 +03:00

lightning_module_template.py

update org paths & convert logos (#685 )

2020-01-20 14:50:31 -05:00

README.md

changes examples to pl_examples for name connflict

2019-10-19 00:41:17 +02:00

README.md

Basic Examples

Use these examples to test how lightning works.

Test on CPU

python cpu_template.py

Train on a single GPU

python gpu_template.py --gpus 1

DataParallel (dp)

Train on multiple GPUs using DataParallel.

python gpu_template.py --gpus 2 --distributed_backend dp

DistributedDataParallel (ddp)

Train on multiple GPUs using DistributedDataParallel

python gpu_template.py --gpus 2 --distributed_backend ddp

DistributedDataParallel+DP (ddp2)

Train on multiple GPUs using DistributedDataParallel + dataparallel. On a single node, uses all GPUs for 1 model. Then shares gradient information across nodes.

python gpu_template.py --gpus 2 --distributed_backend ddp2