fix(trainer): 添加键盘中断处理以保存训练进度

This commit is contained in:
songsenand 2026-04-05 10:38:14 +08:00
parent f838ec9b22
commit c31ec3990f
1 changed files with 5 additions and 2 deletions

View File

@ -865,8 +865,11 @@ def train(
# 开始训练
console.print("\n[bold cyan]开始训练...[/bold cyan]")
console.print(f"开始时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
try:
trainer.train(resume_from=resume_from)
except KeyboardInterrupt:
console.print("[bold green]训练被终止[/bold green]")
trainer.save_checkpoint("interrupted_model.pt")
console.print("[bold green]✓ 训练完成![/bold green]")
console.print(f"结束时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")