warning DDP mode is temporarily unavailable

torchpipe · Nov 7, 2023 · 5b53639 · 5b53639
1 parent 62f34aa
commit 5b53639
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/docs/tools/training.md b/docs/tools/training.md
@@ -73,6 +73,11 @@ In this example, reference code for learning is provided in the torchpipe's exam
   sh train_dp.sh
   ```
 - train_ddp.py # if you use ddp in your code，you can refer to this code.
+
+:::warning DDP mode is temporarily unavailable
+DDP mode may cause the main card's memory load to be higher than other cards.
+:::
+
   ```
   sh train_ddp.sh
   ```

diff --git a/i18n/zh/docusaurus-plugin-content-docs/current/tools/training.md b/i18n/zh/docusaurus-plugin-content-docs/current/tools/training.md
@@ -72,6 +72,11 @@ type: explainer
   sh train_dp.sh
   ```
 - train_ddp.py 如果多卡并行使用了ddp，可以参考这份代码。
+
+:::warning ddp 模式暂时不可用
+ddp模式可能造成主卡显存负载过高于其他卡
+:::
+
   ```
   sh train_ddp.sh
   ```