mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-08-27 12:20:51 +08:00

Files

T

Ir1dXD 5a9afb11cc change print to logging (#457 )

* change print to logging

* always use logging.info

* use f-strings

* update code style

* set logging configs

* remove unused code

2019-11-05 08:43:21 -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

change print to logging (#457 )

2019-11-05 08:43:21 -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