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

Files

T

William FalconandAdrian Wälchli 3cb149f4f4 Removes need to unsqueeze from dp (#1319 )

* removes need to unsqueeze from dp

* removes need to unsqueeze from dp

* fixed examples

* added auto unsqueeze

* added auto unsqueeze

* added auto unsqueeze

* added auto unsqueeze

* Update pytorch_lightning/overrides/data_parallel.py

Co-Authored-By: Adrian Wälchli <adrian.waelchli@students.unibe.ch>

* fixed dp parse

* fixed dp parse

Co-authored-by: Adrian Wälchli <adrian.waelchli@students.unibe.ch>

2020-04-02 11:46:20 -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

Removes need to unsqueeze from dp (#1319 )

2020-04-02 11:46:20 -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