From c31ec3990ff7e7d6d26e60922d5ba7cb4acc575b Mon Sep 17 00:00:00 2001
From: songsenand <songsenand@163.com>
Date: Sun, 5 Apr 2026 10:38:14 +0800
Subject: [PATCH] =?UTF-8?q?fix(trainer):=20=E6=B7=BB=E5=8A=A0=E9=94=AE?=
 =?UTF-8?q?=E7=9B=98=E4=B8=AD=E6=96=AD=E5=A4=84=E7=90=86=E4=BB=A5=E4=BF=9D?=
 =?UTF-8?q?=E5=AD=98=E8=AE=AD=E7=BB=83=E8=BF=9B=E5=BA=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/model/trainer.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/model/trainer.py b/src/model/trainer.py
index 43507c0..61e77f8 100644
--- a/src/model/trainer.py
+++ b/src/model/trainer.py
@@ -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')}")