Skip to content

Commit f4eb231

Browse files
committed
chore(config): Update example_deep_finance configuration and clean up files
- Added a new ignore rule for config file paths in .gitignore - Deleted the automatically generated mcp_finance_tool_generated.json file in example_deep_finance - Refactored the deep_finance.yaml configuration file, adjusting project and experiment names - Reorganized Judge configuration, clarifying openjudge_llm and rm_llm models - Optimized model paths and training parameter configurations, adding parallel and batch processing settings - Adjusted data reading methods and training/validation set path placeholders - Reduced GPU memory usage ratio for rollout to 0.8 - Updated the default save directory path for the trainer to a placeholder variable - Cleaned up unused and commented-out code to improve configuration file conciseness
1 parent 529ae7e commit f4eb231

3 files changed

Lines changed: 26 additions & 41 deletions

File tree

‎.gitignore‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,4 +154,5 @@ site
154154
dump.rdb
155155

156156

157-
tutorial/example_deep_finance/yaml/*
157+
tutorial/example_deep_finance/yaml/*
158+
tutorial/example_deep_finance/config/*

‎tutorial/example_deep_finance/config/mcp_finance_tool_generated.json‎

Lines changed: 0 additions & 10 deletions
This file was deleted.
Lines changed: 24 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,25 @@
11
# ------------------ 主要配置 ------------------
22
ajet:
3-
project_name: ajet
4-
experiment_name: "cc_rm4_res2cit2fai2_30b"
5-
judge_llm: qwen-flash
6-
judge_concurrency: 10
3+
project_name: ajet_deep_finance
4+
experiment_name: "ajet_deep_finance"
5+
# Judge 配置(嵌套结构,对应 self.config.ajet.judge.*)
6+
judge:
7+
openjudge_llm: qwen-flash # OpenJudge 模型
8+
rm_llm: qwen-max # RM Gallery 模型
9+
concurrency: 10 # Judge 并发数
10+
train_ref_ans_path: {{TRAIN_REF_ANS_PATH}} # 训练集 Reference Answer 路径
11+
val_ref_ans_path: {{VAL_REF_ANS_PATH}} # 验证集 Reference Answer 路径
712
# OpenJudge 权重配置
813
report_resolution_weight: 0.2 # 报告质量评估
914
trajectory_faithfulness_weight: 0.2 # 事实准确性评估
1015
citation_audit_weight: 0.2 # 引用审计评估 (覆盖率 + 真实性)
1116
rm_weight: 0.4 # RM Gallery 权重
1217
task_judge:
13-
judge_type: customized_protocol
14-
judge_protocol: tutorial.example_finworld.finworld_judge->DeepFinanceJudgeByOpenJudge
18+
# 使用本地 DeepFinanceJudge 进行评估(解耦远程 env_service)
19+
judge_protocol: tutorial.example_deep_finance.deep_finance_judge->DeepFinanceJudgeByOpenJudge
1520
model:
1621
# ✨✨✨✨ 设置待训练的模型
17-
path: /mnt/data_cpfs/taoshuchang.tsc/models/Qwen3-30B-A3B-Instruct-2507
18-
# path: /mnt/data_cpfs/taoshuchang.tsc/models/Qwen3-8B
22+
path: {{MODEL_PATH}}
1923
trainer_common:
2024
nnodes: 8
2125
n_gpus_per_node: 8
@@ -24,19 +28,20 @@ ajet:
2428
save_freq: 10
2529
test_freq: 2
2630
total_epochs: 200
31+
save_trajectory_as_json_file: True
2732
rollout:
2833
# ✨✨✨✨ 编写并选择Agent
29-
user_workflow: tutorial.example_finworld.finworld->ExampleDeepResearchProtocol
34+
user_workflow: tutorial.example_deep_finance.deep_finance->ExampleDeepResearchProtocol
3035
force_disable_toolcalls: True
3136
enable_oversample: False
3237
tensor_model_parallel_size: 8
3338
num_repeat: 4
3439
max_env_worker: 64 # 增加环境并行数
3540
max_num_seqs: 64 # 增加VLLM并发序列数
36-
max_env_len: 10000
3741
max_response_length_in_one_turn: 8000
3842
max_model_len: 50000
3943
agent_madness_reward: 0.0
44+
compute_madness_checklist: None
4045
multi_turn:
4146
max_steps: 6
4247
interchange_server:
@@ -45,50 +50,39 @@ ajet:
4550
debug_max_parallel: 64 # 增加并行任务数,充分利用GPU
4651
debug_first_n_tasks: 100 # 增加处理的任务数
4752
data:
48-
train_batch_size: 32 # 增加批次大小,适配8卡并行
53+
train_batch_size: 32
4954
max_prompt_length: 8000
5055
max_response_length: 41000
5156

5257
task_reader:
53-
# type: env_service # `env_service` or `dataset_file` or `huggingface_dat_repo` or `finworld`
54-
# === 方案 A: 传统 env_service 模式 ===
55-
# env_service:
56-
# env_type: "finworld"
57-
# env_url: "http://127.0.0.1:8080"
58-
# env_action_preference: code
59-
# training_split: train
60-
# validation_split: val
61-
62-
# === 方案 B: DeepFinance Reader 模式 (数据从 JSON 加载,工具调用走 env_service) ===
63-
type: finworld
64-
finworld:
58+
type: deep_finance # 数据从 JSON 加载并组装 init_messages,工具调用走 env_service
59+
deep_finance:
6560
training:
66-
file_path: /mnt/data_cpfs/taoshuchang.tsc/deepresearch/AgentJet/tutorial/example_finworld/data/finworld_tasks_11171143_cc.json
61+
file_path: {{TRAIN_PATH}}
6762
validation:
68-
file_path: /mnt/data_cpfs/taoshuchang.tsc/deepresearch/AgentJet/tutorial/example_finworld/data/AgentEvolver_query_val.json
69-
# env_service 仍然需要配置(用于工具调用)
63+
file_path: {{VAL_PATH}}
64+
# env_service 仍需配置(用于工具调用)
7065
env_service:
7166
env_type: "finworld"
7267
env_url: "http://127.0.0.1:8080"
7368
env_action_preference: code
7469
trainer:
75-
default_local_dir: "/mnt/data/taoshuchang.tsc/deepresearch/ajet/checkpoints/example_finworld//localths/cc_rm4_res2cit2fai2_30b"
70+
default_local_dir: {{CKPT_SAVE_PATH}}
7671
# resume_mode: disable # 禁用自动恢复,从头开始训练
7772
actor_rollout_ref:
7873
rollout:
7974
tensor_model_parallel_size: 8
80-
gpu_memory_utilization: 0.95
75+
gpu_memory_utilization: 0.8
8176
# ------------------ 不需要修改 ------------------
8277
hydra:
8378
searchpath:
8479
- file://ajet/default_config
8580
- file://ajet/default_config/verl # verl only
86-
- file://external/verl/verl/trainer/config # verl only
8781
- file://ajet/default_config/trinity # trinity only
8882

8983
# ------------------ 不需要修改 ------------------
9084
defaults:
91-
- verl_default # verl inherit 2/2
85+
- verl_default # verl inherit 1/1
9286
- trinity_default # trinity inherit 1/1
9387
- ajet_default
9488
- _self_

0 commit comments

Comments
 (0)