From 9806334517847a51687379384b386d4810571f7a Mon Sep 17 00:00:00 2001
From: carry <luokairui@carry.fit>
Date: Sun, 20 Apr 2025 21:40:46 +0800
Subject: [PATCH] =?UTF-8?q?fix(train=5Fpage):=20=E6=8D=95=E8=8E=B7?=
 =?UTF-8?q?=E8=AE=AD=E7=BB=83=E8=BF=87=E7=A8=8B=E4=B8=AD=E7=9A=84=E5=BC=82?=
 =?UTF-8?q?=E5=B8=B8=E5=B9=B6=E7=BB=88=E6=AD=A2=20TensorBoard=20=E8=BF=9B?=
 =?UTF-8?q?=E7=A8=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在训练过程中添加异常捕获，将异常信息转换为 gr.Error 抛出
- 确保在发生异常时也能终止 TensorBoard 子进程
---
 frontend/train_page.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/frontend/train_page.py b/frontend/train_page.py
index 3cd1903..2b9bafe 100644
--- a/frontend/train_page.py
+++ b/frontend/train_page.py
@@ -82,6 +82,8 @@ def train_page():
                             dataset, new_training_dir, 
                             learning_rate, per_device_train_batch_size, epoch,
                             save_steps, lora_rank)
+            except Exception as e:
+                raise gr.Error(str(e))
             finally:
                 # 确保训练结束后终止 TensorBoard 子进程
                 tensorboard_process.terminate()