github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-01 17:17:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CI] fix some spelling errors (#3707)

Browse Source 
* fix spelling error with examples/comminity/

* fix spelling error with tests/

* fix some spelling error with tests/ colossalai/ etc.
This commit is contained in:
digger-yu
2023-05-10 17:12:03 +08:00
committed by GitHub
parent f7361ee1bd
commit b7141c36dd
17 changed files with 51 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/tutorial/new_api/cifar_resnet/train.py
View File

@@ -28,7 +28,7 @@ LEARNING_RATE = 1e-3
def build_dataloader(batch_size: int, coordinator: DistCoordinator, plugin: DPPluginBase):
# trainsform
# transform
transform_train = transforms.Compose(
[transforms.Pad(4),
transforms.RandomHorizontalFlip(),

6
examples/tutorial/new_api/cifar_vit/train.py
View File

@@ -25,7 +25,7 @@ from colossalai.utils import get_current_device
# Prepare Hyperparameters
# ==============================
NUM_EPOCHS = 60
WARMUP_EPOCSH = 5
WARMUP_EPOCHS = 5
LEARNING_RATE = 1e-3
@@ -37,7 +37,7 @@ def vit_cifar(**kwargs):
def build_dataloader(batch_size: int, coordinator: DistCoordinator, plugin: DPPluginBase):
# trainsform
# transform
transform_train = transforms.Compose([
transforms.RandomCrop(32, padding=4),
transforms.RandomHorizontalFlip(),
@@ -177,7 +177,7 @@ def main():
optimizer = HybridAdam(model.parameters(), lr=LEARNING_RATE)
# lr scheduler
lr_scheduler = LinearWarmupLR(optimizer, NUM_EPOCHS, WARMUP_EPOCSH)
lr_scheduler = LinearWarmupLR(optimizer, NUM_EPOCHS, WARMUP_EPOCHS)
# ==============================
# Boost with ColossalAI