From a27a1ab0794dd0224059898373943ad43cf3e3a7 Mon Sep 17 00:00:00 2001 From: carry Date: Tue, 15 Apr 2025 00:12:09 +0800 Subject: [PATCH] =?UTF-8?q?refactor(frontend):=20=E9=87=8D=E6=9E=84?= =?UTF-8?q?=E8=AE=AD=E7=BB=83=E9=A1=B5=E9=9D=A2=E5=B8=83=E5=B1=80=E5=B9=B6?= =?UTF-8?q?=E4=BC=98=E5=8C=96=E7=94=A8=E6=88=B7=E7=95=8C=E9=9D=A2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 调整数据集下拉框布局位置 - 新增超参数输入组件 - 修改训练日志输出框标签为"训练状态" - 添加 TensorBoard 可视化 iframe 显示框 --- frontend/train_page.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) diff --git a/frontend/train_page.py b/frontend/train_page.py index 2acbbd5..bcfd986 100644 --- a/frontend/train_page.py +++ b/frontend/train_page.py @@ -16,17 +16,15 @@ def train_page(): # 获取数据集列表并设置初始值 datasets_list = [str(ds["name"]) for ds in get_datasets().all()] initial_dataset = datasets_list[0] if datasets_list else None - - dataset_dropdown = gr.Dropdown( - choices=datasets_list, - value=initial_dataset, # 设置初始选中项 - label="选择数据集", - allow_custom_value=True, - interactive=True - ) - with gr.Row(): with gr.Column(scale=1): + dataset_dropdown = gr.Dropdown( + choices=datasets_list, + value=initial_dataset, # 设置初始选中项 + label="选择数据集", + allow_custom_value=True, + interactive=True + ) # 新增超参数输入组件 learning_rate_input = gr.Number(value=2e-4, label="学习率") per_device_train_batch_size_input = gr.Number(value=1, label="batch size", precision=0) @@ -37,7 +35,7 @@ def train_page(): train_button = gr.Button("开始微调") # 训练状态输出 - output = gr.Textbox(label="训练日志", interactive=False) + output = gr.Textbox(label="训练状态", interactive=False) with gr.Column(scale=3): # 新增 TensorBoard iframe 显示框 tensorboard_iframe = gr.HTML(label="TensorBoard 可视化")