File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -55,14 +55,14 @@ function check_whl {
5555function 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
6868function 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
You can’t perform that action at this time.
0 commit comments