ViperEkura
/
AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: 修复参数问题

This commit is contained in:
ViperEkura 2026-04-03 23:34:21 +08:00
parent c0e0e6afd9
commit 70d52935f0
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/tools/server.py
View File

@ -13,7 +13,7 @@ def main():
"--port", type=int, default=8000, help="Port number (default: 8000)" "--port", type=int, default=8000, help="Port number (default: 8000)"
) )
parser.add_argument( 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( parser.add_argument(
"--param-path", "--param-path",

2
scripts/tools/train.py
View File

@ -225,7 +225,7 @@ def train(
scheduler_fn = partial( scheduler_fn = partial(
create_scheduler, create_scheduler,
**{ **{
"scheduler": "cosine", "schedule_type": "cosine",
"warmup_steps": warmup_steps, "warmup_steps": warmup_steps,
"lr_decay_steps": toltal_steps - warmup_steps, "lr_decay_steps": toltal_steps - warmup_steps,
}, },