github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-10 13:30:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

[doc] update amp document

Browse Source
This commit is contained in:
Mingyan Jiang
2023-05-23 13:11:03 +08:00
committed by jiangmingyan
parent 9265f2d4d7
commit 8c62e50dbb
5 changed files with 24 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/source/en/features/mixed_precision_training.md
View File

@@ -1,4 +1,4 @@
# Auto Mixed Precision Training
# Auto Mixed Precision Training (Outdated)
Author: Chuanrui Wang, Shenggui Li, Yongbin Li
@@ -365,3 +365,4 @@ Use the following command to start the training scripts. You can change `--nproc
```python
python -m torch.distributed.launch --nproc_per_node 4 --master_addr localhost --master_port 29500 train_with_engine.py --config config/config_AMP_torch.py
```
<!-- doc-test-command: torchrun --standalone --nproc_per_node=1 mixed_precision_training.py -->