fix(trainer): 添加键盘中断处理以保存训练进度
This commit is contained in:
parent
f838ec9b22
commit
c31ec3990f
|
|
@ -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')}")
|
||||
|
||||
trainer.train(resume_from=resume_from)
|
||||
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')}")
|
||||
|
|
|
|||
Loading…
Reference in New Issue