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 可视化")