chore: 修改错误拼写

2026-04-06 10:37:19 +08:00
parent ace8f6ee68
commit f2ffdf60d0
4 changed files with 14 additions and 13 deletions
@@ -234,13 +234,13 @@ def train(
        },
    )

-    toltal_steps = len(dataset) * n_epoch // (batch_size * nprocs)
+    total_steps = len(dataset) * n_epoch // (batch_size * nprocs)
    scheduler_fn = partial(
        create_scheduler,
        **{
            "schedule_type": "cosine",
            "warmup_steps": warmup_steps,
-            "lr_decay_steps": toltal_steps - warmup_steps,
+            "lr_decay_steps": total_steps - warmup_steps,
        },
    )