wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-24 12:41:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated docs

Browse Source
This commit is contained in:
William Falcon committed 2019-07-21 08:35:29 -04:00
1 parent f6b98fe74f
commit 7ac344e43a
1 file changed
+5 -1
Download Patch File Download Diff File Expand all files Collapse all files
docs/Trainer/Distributed training.md
+5 -1
View File
@@ -63,7 +63,11 @@ In this setting, the model will run on all 8 GPUs at once using DataParallel und
# os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3,4,5,6,7"
trainer = Trainer(gpus=[0,1,2,3,4,5,6,7])
# to use DataParallel (default)
trainer = Trainer(gpus=[0,1,2,3,4,5,6,7], distributed_backend='dp')
# RECOMMENDED use DistributedDataParallel
trainer = Trainer(gpus=[0,1,2,3,4,5,6,7], distributed_backend='ddp')
```
---