Deployed 7ac344e with MkDocs version: 1.0.4

2026-09-09 11:32:07 +08:00 · 2019-07-21 07:35:40 -05:00
parent 6eea1eac31
commit 3536c7c427
3 changed files with 6 additions and 2 deletions
@@ -657,7 +657,11 @@ In this setting, the model will run on all 8 GPUs at once using DataParallel und
 # os.environ[&quot;CUDA_VISIBLE_DEVICES&quot;] = &quot;0,1,2,3,4,5,6,7&quot;


-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')
 </code></pre>

 <hr />