Skip to content

Commit 818a4f7

Browse files
committed
fix(deep_finance): update environment variables and training launch options
1 parent 6f0c420 commit 818a4f7

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

tutorial/example_deep_finance/deep_finance.sh

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
2-
set -e
2+
set -e
33
#===============================================================================
44
# 1. 配置区域 - 用户只需修改这里
55
#===============================================================================
@@ -22,7 +22,7 @@ TRAIN_BATCH_SIZE=32 # 训练batchsize
2222
NUM_STEPS=6 # 每个样本step轮数
2323
DEEPFINANCE_TOOL_RESULT_MAX_CHARS=10000
2424

25-
# 主目录 (按需更改)
25+
# 主目录(需要更改)
2626
export AJET_ROOT="/mnt/data_cpfs/taoshuchang.tsc/deepresearch/AgentJet_new"
2727

2828
NNODES=${WORLD_SIZE}
@@ -105,7 +105,7 @@ export DEEPFINANCE_MCP_CONFIG DEEPFINANCE_TOOL_RESULT_MAX_CHARS
105105
# 其他服务配置
106106
HF_ENDPOINT="https://hf-mirror.com"
107107
ES_HOSTS="http://11.160.132.46:8200"
108-
export HF_ENDPOINT ES_HOSTS
108+
export HF_ENDPOINT ES_HOSTS
109109

110110
# log 文件位置
111111
CURRENT_TIME=$(date "+%Y%m%d_%H%M%S")
@@ -155,6 +155,8 @@ export NCCL_ASYNC_ERROR_HANDLING=1
155155

156156
export PYTHONPATH="${AJET_ROOT}:${PYTHONPATH}"
157157
export RAY_CLUSTER_MODE="multi_node"
158+
export DEEPFINANCE_PATH="${ENV_SERVICE_ROOT}" # AgentJet 内部可能使用此路径
159+
export DEEPFINANCE_SCRIPT="source /mnt/data/taoshuchang.tsc/anaconda3/etc/profile.d/conda.sh && conda activate finworld_1209 && cd ${ENV_SERVICE_ROOT} && DEEPFINANCE_TOOL_RESULT_MAX_CHARS=${DEEPFINANCE_TOOL_RESULT_MAX_CHARS} DEEPFINANCE_MCP_CONFIG=${DEEPFINANCE_MCP_CONFIG} CACHE_TYPE=${CACHE_TYPE} MONGO_URI=${MONGO_URI} MONGO_DB_NAME=${MONGO_DB_NAME} MONGO_COLLECTION_NAME=${MONGO_COLLECTION_NAME} python -m env_service.env_service --env finworld --portal 0.0.0.0 --port 8080"
158160

159161

160162
#===============================================================================
@@ -201,11 +203,12 @@ if [[ $HOSTNAME == *"-master-"* ]]; then
201203

202204
# 启动训练任务(最核心)
203205
python ajet/launcher.py \
206+
--with-deepfinance \
204207
--conf ${CONFIG_FILE} \
205208
--backbone="verl" \
206209
--prefix=${env_log_prefix} \
207210
2>&1 | tee ${TRAIN_LOG}
208-
211+
209212

210213
#===============================================================================
211214
# 6.2 Worker 节点启动流程
@@ -217,4 +220,4 @@ else
217220
ray stop || true
218221
ray start --address $MASTER_ADDR:6379 --num-gpus 8
219222
while true; do sleep 60; done
220-
fi
223+
fi

0 commit comments

Comments
 (0)