Skip to content

Commit 75cfd4f

Browse files
authored
add resnet_bs64_fp16 for model_benchmark_ci (#1572)
* add resnet_bs64_fp16 for model_benchmark_ci,test=document_fix * add resnet_bs64_fp16 for model_benchmark_ci,test=document_fix
1 parent f900bef commit 75cfd4f

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

scripts/benchmark_ci/model_ci.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ export FLAGS_call_stack_level=2
1515
#run models
1616
cd ${BENCHMARK_ROOT}/scripts/benchmark_ci
1717
#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'
18-
model_list='ResNet50_bs32_dygraph deeplabv3_bs4_fp32 bert_base_seqlen128_fp32_bs32 yolov3_bs8 ppyolov2_bs6'
18+
model_list='ResNet50_bs32_dygraph ResNet50_bs64_fp16_dygraph deeplabv3_bs4_fp32 bert_base_seqlen128_fp32_bs32 yolov3_bs8 ppyolov2_bs6'
1919
source run_models.sh
2020
for model in ${model_list}
2121
do

scripts/benchmark_ci/run_models.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ ResNet50_bs64_fp16_dygraph(){
2222
cur_model_path=${BENCHMARK_ROOT}/PaddleClas
2323
cd ${cur_model_path}
2424
pip install -r requirements.txt
25-
#install dali
25+
# install dali
2626
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda110
2727
# Prepare data
2828
ln -s ${data_path}/imagenet100_data/ ${cur_model_path}/dataset

0 commit comments

Comments
 (0)