diff --git a/docs/zh_cn/tutorials/runner.md b/docs/zh_cn/tutorials/runner.md index 497130cbbc220b7c9bc6c18b076b5d49cedf3ad6..01a311294302e3ade42b359dec25d30b40709a58 100644 --- a/docs/zh_cn/tutorials/runner.md +++ b/docs/zh_cn/tutorials/runner.md @@ -42,9 +42,9 @@ class Network(BaseModel): model = Network() # 构建优化器 -optimzier = torch.optim.SGD(model.parameters(), lr=0.01, momentum=0.9) +optimizer = torch.optim.SGD(model.parameters(), lr=0.01, momentum=0.9) # 构建å‚数调度器用于调整å¦ä¹ 率 -lr_scheduler = MultiStepLR(milestones=[2], by_epoch=True) +lr_scheduler = MultiStepLR(optimizer, milestones=[2], by_epoch=True) # 构建手写数å—识别 (MNIST) æ•°æ®é›† train_dataset = datasets.MNIST(root="MNIST", download=True, train=True, transform=transforms.ToTensor()) # 构建数æ®åŠ 载器 @@ -58,7 +58,7 @@ from mmengine.runner import Runner # è®ç»ƒç›¸å…³å‚数设置,按轮次è®ç»ƒï¼Œè®ç»ƒ3è½® -train_cfg = dict(by_epoch=True, max_epoch=3) +train_cfg = dict(by_epoch=True, max_epochs=3) # åˆå§‹åŒ–执行器 runner = Runner(model,