[doc] moved doc test command to bottom (#3075)

This commit is contained in:
Frank Lee
2023-03-09 18:10:45 +08:00
committed by GitHub
parent 91ccf97514
commit 416a50dbd7
5 changed files with 14 additions and 7 deletions

View File

@@ -1,4 +1,3 @@
<!-- doc-test-command: torchrun --standalone --nproc_per_node=1 nvme_offload.py -->
# NVMe offload
作者: Hongxin Liu
@@ -247,3 +246,6 @@ NVME 卸载节省了大约 294 MB 内存。注意使用 Gemini 的 `pin_memory`
{{ autodoc:colossalai.nn.optimizer.HybridAdam }}
{{ autodoc:colossalai.nn.optimizer.CPUAdam }}
<!-- doc-test-command: torchrun --standalone --nproc_per_node=1 nvme_offload.py -->