pull/3810/head
jiangmingyan 2023-05-23 17:50:30 +08:00
parent c425a69d52
commit 8aa1fb2c7f
2 changed files with 2 additions and 2 deletions

View File

@ -246,6 +246,6 @@ for epoch in range(NUM_EPOCHS):
Use the following command to start the training scripts. You can change `--nproc_per_node` to use a different number of GPUs.
```shell
colossalai run --nproc_per_node 1 train.py --config config/config.py
colossalai run --nproc_per_node 1 train.py
```
<!-- doc-test-command: torchrun --standalone --nproc_per_node=1 mixed_precision_training_with_booster.py -->

View File

@ -230,6 +230,6 @@ for epoch in range(NUM_EPOCHS):
使用下列命令启动训练脚本,你可以改变 `--nproc_per_node` 以使用不同数量的 GPU。
```shell
colossalai run --nproc_per_node 1 train.py --config config/config.py
colossalai run --nproc_per_node 1 train.py
```
<!-- doc-test-command: torchrun --standalone --nproc_per_node=1 mixed_precision_training_with_booster.py -->