Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix benchmark dir and add CUDA_DEVICE_MAX_CONNECTIONS to qwen #8678

Merged
merged 17 commits into from
Jun 28, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -203,7 +203,7 @@ export FLAGS_cudnn_deterministic=True
export FLAGS_cudnn_deterministic=1
export FLAGS_embedding_deterministic=1

export PYTHONPATH="../../../PaddleNLP/"
export PYTHONPATH="../../PaddleNLP/"
source ${BENCHMARK_ROOT}/scripts/run_model.sh # 在该脚本中会对符合benchmark规范的log使用analysis.py 脚本进行性能数据解析;如果不联调只想要产出训练log可以注掉本行,提交时需打开
_set_params $@
#_train # 如果只产出训练log,不解析,可取消注释
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -230,6 +230,7 @@ function _train(){
}

export PYTHONPATH=$(dirname "$PWD"):$PYTHONPATH
export CUDA_DEVICE_MAX_CONNECTIONS=1

source ${BENCHMARK_ROOT}/scripts/run_model.sh # 在该脚本中会对符合benchmark规范的log使用analysis.py 脚本进行性能数据解析;如果不联调只想要产出训练log可以注掉本行,提交时需打开
_set_params $@
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,4 +25,4 @@ python3 -m pip install --upgrade paddlenlp
cd ../legacy/model_zoo/moe/dygraph/
rm -rf data
mkdir data && cd data
wget https://bj.bcebos.com/paddlenlp/models/transformers/gpt2/train.data.json_ids.npz
wget https://bj.bcebos.com/paddlenlp/models/transformers/gpt2/train.data.json_ids.npz
Loading