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

[hotfix] quick fixes to make legacy tutorials runnable (#5559)

Browse Source 
Co-authored-by: Edenzzzz <wtan45@wisc.edu>
This commit is contained in:
Edenzzzz
2024-04-07 12:06:27 +08:00
committed by GitHub
parent 8e412a548e
commit 15055f9a36
8 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/tutorial/large_batch_optimizer/train.py
View File

@@ -37,14 +37,14 @@ class DummyDataloader:
def main():
# initialize distributed setting
parser = colossalai.get_default_parser()
parser = colossalai.legacy.get_default_parser()
parser.add_argument(
"--optimizer", choices=["lars", "lamb"], help="Choose your large-batch optimizer", required=True
)
args = parser.parse_args()
# launch from torch
colossalai.launch_from_torch(config=args.config)
colossalai.legacy.launch_from_torch(config=args.config)
# get logger
logger = get_dist_logger()
@@ -73,7 +73,7 @@ def main():
)
# initialize
engine, train_dataloader, test_dataloader, _ = colossalai.initialize(
engine, train_dataloader, test_dataloader, _ = colossalai.legacy.initialize(
model=model,
optimizer=optimizer,
criterion=criterion,