fix: 修复参数问题

2026-04-03 23:34:21 +08:00
parent c0e0e6afd9
commit 70d52935f0
2 changed files with 2 additions and 2 deletions
@@ -225,7 +225,7 @@ def train(
    scheduler_fn = partial(
        create_scheduler,
        **{
-            "scheduler": "cosine",
+            "schedule_type": "cosine",
            "warmup_steps": warmup_steps,
            "lr_decay_steps": toltal_steps - warmup_steps,
        },