support report custom args
Former-commit-id: d41254c40a1c5cacf9377096adb27efa9bdb79ea
This commit is contained in:
hiyouga
@@ -273,21 +273,23 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
|
||||
with gr.Accordion(open=False) as swanlab_tab:
|
||||
with gr.Row():
|
||||
use_swanlab = gr.Checkbox()
|
||||
swanlab_project = gr.Textbox(value="llamafactory", placeholder="Project name", interactive=True)
|
||||
swanlab_experiment_name = gr.Textbox(value="", placeholder="Experiment name", interactive=True)
|
||||
swanlab_workspace = gr.Textbox(value="", placeholder="Workspace name", interactive=True)
|
||||
swanlab_api_key = gr.Textbox(value="", placeholder="API key", interactive=True)
|
||||
swanlab_mode = gr.Dropdown(choices=["cloud", "local", "disabled"], value="cloud", interactive=True)
|
||||
swanlab_project = gr.Textbox(value="llamafactory")
|
||||
swanlab_run_name = gr.Textbox()
|
||||
swanlab_workspace = gr.Textbox()
|
||||
swanlab_api_key = gr.Textbox()
|
||||
swanlab_mode = gr.Dropdown(choices=["cloud", "local"], value="cloud")
|
||||
|
||||
input_elems.update({use_swanlab, swanlab_api_key, swanlab_project, swanlab_workspace, swanlab_experiment_name, swanlab_mode})
|
||||
input_elems.update(
|
||||
{use_swanlab, swanlab_project, swanlab_run_name, swanlab_workspace, swanlab_api_key, swanlab_mode}
|
||||
)
|
||||
elem_dict.update(
|
||||
dict(
|
||||
swanlab_tab=swanlab_tab,
|
||||
use_swanlab=use_swanlab,
|
||||
swanlab_api_key=swanlab_api_key,
|
||||
swanlab_project=swanlab_project,
|
||||
swanlab_run_name=swanlab_run_name,
|
||||
swanlab_workspace=swanlab_workspace,
|
||||
swanlab_experiment_name=swanlab_experiment_name,
|
||||
swanlab_api_key=swanlab_api_key,
|
||||
swanlab_mode=swanlab_mode,
|
||||
)
|
||||
)
|
||||
|
||||
@@ -1385,86 +1385,85 @@ LOCALES = {
|
||||
"info": "SwanLab를 사용하여 실험을 추적하고 시각화합니다.",
|
||||
},
|
||||
},
|
||||
"swanlab_api_key": {
|
||||
"en": {
|
||||
"label": "API Key(optional)",
|
||||
"info": "API key for SwanLab. Once logged in, no need to login again in the programming environment.",
|
||||
},
|
||||
"ru": {
|
||||
"label": "API ключ(Необязательный)",
|
||||
"info": "API ключ для SwanLab. После входа в программное окружение, нет необходимости входить снова.",
|
||||
},
|
||||
"zh": {
|
||||
"label": "API密钥(选填)",
|
||||
"info": "用于在编程环境登录SwanLab，已登录则无需填写。",
|
||||
},
|
||||
"ko": {
|
||||
"label": "API 키(선택 사항)",
|
||||
"info": "SwanLab의 API 키. 프로그래밍 환경에 로그인한 후 다시 로그인할 필요가 없습니다.",
|
||||
},
|
||||
},
|
||||
"swanlab_project": {
|
||||
"en": {
|
||||
"label": "Project(optional)",
|
||||
"label": "SwanLab project",
|
||||
},
|
||||
"ru": {
|
||||
"label": "Проект(Необязательный)",
|
||||
"label": "SwanLab Проект",
|
||||
},
|
||||
"zh": {
|
||||
"label": "项目(选填)",
|
||||
"label": "SwanLab 项目名",
|
||||
},
|
||||
"ko": {
|
||||
"label": "프로젝트(선택 사항)",
|
||||
"label": "SwanLab 프로젝트",
|
||||
},
|
||||
},
|
||||
"swanlab_run_name": {
|
||||
"en": {
|
||||
"label": "SwanLab experiment name (optional)",
|
||||
},
|
||||
"ru": {
|
||||
"label": "SwanLab Имя эксперимента (опционально)",
|
||||
},
|
||||
"zh": {
|
||||
"label": "SwanLab 实验名（非必填）",
|
||||
},
|
||||
"ko": {
|
||||
"label": "SwanLab 실험 이름 (선택 사항)",
|
||||
},
|
||||
},
|
||||
"swanlab_workspace": {
|
||||
"en": {
|
||||
"label": "Workspace(optional)",
|
||||
"info": "Workspace for SwanLab. If not filled, it defaults to the personal workspace.",
|
||||
|
||||
"label": "SwanLab workspace (optional)",
|
||||
"info": "Workspace for SwanLab. Defaults to the personal workspace.",
|
||||
},
|
||||
"ru": {
|
||||
"label": "Рабочая область(Необязательный)",
|
||||
"label": "SwanLab Рабочая область (опционально)",
|
||||
"info": "Рабочая область SwanLab, если не заполнено, то по умолчанию в личной рабочей области.",
|
||||
},
|
||||
"zh": {
|
||||
"label": "Workspace(选填)",
|
||||
"info": "SwanLab组织的工作区，如不填写则默认在个人工作区下",
|
||||
"label": "SwanLab 工作区（非必填）",
|
||||
"info": "SwanLab 的工作区，默认在个人工作区下。",
|
||||
},
|
||||
"ko": {
|
||||
"label": "작업 영역(선택 사항)",
|
||||
"label": "SwanLab 작업 영역 (선택 사항)",
|
||||
"info": "SwanLab 조직의 작업 영역, 비어 있으면 기본적으로 개인 작업 영역에 있습니다.",
|
||||
},
|
||||
},
|
||||
"swanlab_experiment_name": {
|
||||
"swanlab_api_key": {
|
||||
"en": {
|
||||
"label": "Experiment name (optional)",
|
||||
"label": "SwanLab API key (optional)",
|
||||
"info": "API key for SwanLab.",
|
||||
},
|
||||
"ru": {
|
||||
"label": "Имя эксперимента(Необязательный)",
|
||||
"label": "SwanLab API ключ (опционально)",
|
||||
"info": "API ключ для SwanLab.",
|
||||
},
|
||||
"zh": {
|
||||
"label": "实验名(选填) ",
|
||||
"label": "SwanLab API密钥（非必填）",
|
||||
"info": "用于在编程环境登录 SwanLab，已登录则无需填写。",
|
||||
},
|
||||
"ko": {
|
||||
"label": "실험 이름(선택 사항)",
|
||||
"label": "SwanLab API 키 (선택 사항)",
|
||||
"info": "SwanLab의 API 키.",
|
||||
},
|
||||
},
|
||||
"swanlab_mode": {
|
||||
"en": {
|
||||
"label": "Mode",
|
||||
"info": "Cloud or offline version.",
|
||||
"label": "SwanLab mode",
|
||||
"info": "Cloud or offline version.",
|
||||
},
|
||||
"ru": {
|
||||
"label": "Режим",
|
||||
"label": "SwanLab Режим",
|
||||
"info": "Версия в облаке или локальная версия.",
|
||||
},
|
||||
"zh": {
|
||||
"label": "模式",
|
||||
"info": "云端版或离线版",
|
||||
"label": "SwanLab 模式",
|
||||
"info": "使用云端版或离线版 SwanLab。",
|
||||
},
|
||||
"ko": {
|
||||
"label": "모드",
|
||||
"label": "SwanLab 모드",
|
||||
"info": "클라우드 버전 또는 오프라인 버전.",
|
||||
},
|
||||
},
|
||||
|
||||
@@ -231,12 +231,11 @@ class Runner:
|
||||
|
||||
# swanlab config
|
||||
if get("train.use_swanlab"):
|
||||
args["swanlab_api_key"] = get("train.swanlab_api_key")
|
||||
args["swanlab_project"] = get("train.swanlab_project")
|
||||
args["swanlab_run_name"] = get("train.swanlab_run_name")
|
||||
args["swanlab_workspace"] = get("train.swanlab_workspace")
|
||||
args["swanlab_experiment_name"] = get("train.swanlab_experiment_name")
|
||||
args["swanlab_api_key"] = get("train.swanlab_api_key")
|
||||
args["swanlab_mode"] = get("train.swanlab_mode")
|
||||
|
||||
|
||||
# eval config
|
||||
if get("train.val_size") > 1e-6 and args["stage"] != "ppo":
|
||||
|
||||
Reference in New Issue
Block a user