mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-02 12:40:56 +08:00

Files

T

Jirka Borovec 45d671a4a8 CI: split tests-examples (#990 )

* CI: split tests-examples

* tests without template

* comment depends

* CircleCI typo

* add doctest

* update test req.

* CI tests

* setup macOS

* longer train

* lover pred acc

* fix model

* rename default model

* lower tests acc

* typo

* imports

* fix test optimizer

* update calls

* fix Win

* lower Drone image

* fix call

* pytorch image

* fix test

* add dev image

* add dev image

* update image

* drone volume

* lint

* update test notes

* rename tests/models >> tests/base

* group models

* conftest

* optim imports

* typos

* fix import

* fix tests

* install AMP

* tests

* fix import

2020-03-25 07:46:27 -04:00

__init__.py

changes examples to pl_examples for name connflict

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

cpu_template.py

fix max_epochs setup in basic example (#1105 )

2020-03-12 12:42:52 -04:00

gpu_template.py

fix max_epochs setup in basic example (#1105 )

2020-03-12 12:42:52 -04:00

lightning_module_template.py

CI: split tests-examples (#990 )

2020-03-25 07:46:27 -04: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