Skip to content

Commit 1eaff94

Browse files
committed
add more models for model_benchmark_ci,test=document_fix
1 parent d1a33bc commit 1eaff94

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tools/test_ci_model_benchmark.sh

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,14 +55,14 @@ function check_whl {
5555
function compile_install_paddle {
5656
export CUDA_ARCH_NAME=${CUDA_ARCH_NAME:-Auto}
5757
export PY_VERSION=3.7
58-
export WITH_DISTRIBUTE=OFF
58+
export WITH_DISTRIBUTE=ON
5959
export WITH_GPU=ON
6060
export WITH_TENSORRT=OFF
6161
export WITH_TESTING=OFF
6262
export WITH_UNITY_BUILD=ON
6363
check_whl
6464
cd /workspace/Paddle
65-
git clone --recurse-submodules=PaddleClas --recurse-submodules=PaddleNLP https://github.com/paddlepaddle/benchmark.git
65+
git clone --recurse-submodules=PaddleClas --recurse-submodules=PaddleNLP --recurse-submodules=PaddleDetection --recurse-submodules=PaddleVideo --recurse-submodules=PaddleSeg --recurse-submodules=PaddleGAN --recurse-submodules=PaddleOCR --recurse-submodules=models https://github.com/paddlepaddle/benchmark.git
6666
}
6767

6868
function prepare_data {
@@ -83,7 +83,8 @@ function run_model_benchmark {
8383
cd /workspace/Paddle
8484
pip install build/pr_whl/*.whl
8585
cd ${cache_dir}/benchmark_data
86-
export data_path=${cache_dir}/benchmark_data/dataset
86+
export data_path=${cfs_dir}/model_dataset/model_benchmark_data
87+
export prepare_path=${cfs_dir}/model_dataset/model_benchmark_prepare
8788
export BENCHMARK_ROOT=/workspace/Paddle/benchmark
8889
cd ${BENCHMARK_ROOT}/scripts/benchmark_ci
8990
bash model_ci.sh

0 commit comments

Comments
 (0)