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

fix typo docs/ (#4033)

Browse Source
This commit is contained in:
digger yu
2023-06-28 15:30:30 +08:00
committed by GitHub
parent 2d40759a53
commit 769cddcb2c
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/zh-Hans/features/mixed_precision_training.md
View File

@@ -303,7 +303,7 @@ colossalai.launch_from_torch(config=args.config)
# build loss
criterion = torch.nn.CrossEntropyLoss()
# lr_scheduelr
# lr_scheduler
lr_scheduler = LinearWarmupLR(optimizer, warmup_steps=50, total_steps=gpc.config.NUM_EPOCHS)
```

2
docs/source/zh-Hans/features/mixed_precision_training_with_booster.md
View File

@@ -181,7 +181,7 @@ optimizer = torch.optim.SGD(model.parameters(), lr=1e-2, weight_decay=0.1)
# build loss
criterion = torch.nn.CrossEntropyLoss()
# lr_scheduelr
# lr_scheduler
lr_scheduler = LinearWarmupLR(optimizer, warmup_steps=50, total_steps=NUM_EPOCHS)
```