fix: 修复参数问题
This commit is contained in:
parent
c0e0e6afd9
commit
70d52935f0
|
|
@ -13,7 +13,7 @@ def main():
|
|||
"--port", type=int, default=8000, help="Port number (default: 8000)"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--reload", action="store_true", help="Enable auto‑reload for development"
|
||||
"--reload", action="store_true", help="Enable auto-reload for development"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--param-path",
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
},
|
||||
|
|
|
|||
Loading…
Reference in New Issue