File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -15,7 +15,7 @@ export FLAGS_call_stack_level=2
15
15
# run models
16
16
cd ${BENCHMARK_ROOT} /scripts/benchmark_ci
17
17
# 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'
19
19
source run_models.sh
20
20
for model in ${model_list}
21
21
do
Original file line number Diff line number Diff line change @@ -22,7 +22,7 @@ ResNet50_bs64_fp16_dygraph(){
22
22
cur_model_path=${BENCHMARK_ROOT} /PaddleClas
23
23
cd ${cur_model_path}
24
24
pip install -r requirements.txt
25
- # install dali
25
+ # install dali
26
26
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda110
27
27
# Prepare data
28
28
ln -s ${data_path} /imagenet100_data/ ${cur_model_path} /dataset
You can’t perform that action at this time.
0 commit comments