fix test doc

hpcaitech · May 13, 2024 · 524a16f · 524a16f
1 parent 208817a
commit 524a16f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source/en/features/distributed_optimizers.md b/docs/source/en/features/distributed_optimizers.md
@@ -138,4 +138,4 @@ optim = DistGaloreAwamW(
   </tr>
 </table>
 
-<!-- doc-test-command: python test_dist_optim.py  -->
+<!-- doc-test-command: colossalai run --nproc_per_node 4 distributed_optimizers.py  -->
diff --git a/docs/source/zh-Hans/features/distributed_optimizers.md b/docs/source/zh-Hans/features/distributed_optimizers.md
@@ -138,4 +138,4 @@ optim = DistGaloreAwamW(
   </tr>
 </table>
 
-<!-- doc-test-command: python test_dist_optim.py  -->
+<!-- doc-test-command: colossalai run --nproc_per_node 4 distributed_optimizers.py  -->