Skip to content

add resnet_bs64_fp16 for model_benchmark_ci #1572

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

Merged
merged 2 commits into from
Jan 17, 2023
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
2 changes: 1 addition & 1 deletion scripts/benchmark_ci/model_ci.sh
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ export FLAGS_call_stack_level=2
#run models
cd ${BENCHMARK_ROOT}/scripts/benchmark_ci
#model_list='ResNet50_bs32_dygraph ResNet50_bs32 bert_base_seqlen128_fp32_bs32 transformer_base_bs4096_amp_fp16 yolov3_bs8 TSM_bs16 deeplabv3_bs4_fp32 CycleGAN_bs1 mask_rcnn_bs1 PPOCR_mobile_2_bs8 seq2seq_bs128'
model_list='ResNet50_bs32_dygraph deeplabv3_bs4_fp32 bert_base_seqlen128_fp32_bs32 yolov3_bs8 ppyolov2_bs6'
model_list='ResNet50_bs32_dygraph ResNet50_bs64_fp16_dygraph deeplabv3_bs4_fp32 bert_base_seqlen128_fp32_bs32 yolov3_bs8 ppyolov2_bs6'
source run_models.sh
for model in ${model_list}
do
Expand Down
2 changes: 1 addition & 1 deletion scripts/benchmark_ci/run_models.sh
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ ResNet50_bs64_fp16_dygraph(){
cur_model_path=${BENCHMARK_ROOT}/PaddleClas
cd ${cur_model_path}
pip install -r requirements.txt
#install dali
# install dali
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda110
# Prepare data
ln -s ${data_path}/imagenet100_data/ ${cur_model_path}/dataset
Expand Down