p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!5334 update the FP16 model copy rules for CI 

Merge pull request !5334 from mengchunyang/mm
This commit is contained in:
mindspore-ci-bot 2020-08-27 18:48:09 +08:00 committed by Gitee
parent da8d04e8bd 7dd18c0d2e
commit a48bbf49d6
2 changed files with 13 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/lite/test/models_fp16.cfg
View File

@ -1,7 +1,7 @@
detect-deeper-halfdeeper-mbv1-shortcut-400-400_nopostprocess_simplified.fp16
model_emotions_0727_nosoftmax.fp16
model_emotions_0727_nosoftmax.tflite.fp16
mtk_isface.fp16
mtk_landmark.fp16
mtk_pose_tuku.fp16
mtk_age_gender.fp16
mtk_model_face_dress.fp16
mtk_age_gender.tflite.fp16
mtk_model_face_dress.tflite.fp16

21
mindspore/lite/test/run_benchmark_nets.sh
View File

@ -403,17 +403,6 @@ rm -rf ${basepath}/ms_models
mkdir -p ${basepath}/ms_models
ms_models_path=${basepath}/ms_models
echo "start convert models ... :"
# Copy fp16 ms models:
while read line; do
model_name=${line}
if [[ $model_name == \#* ]]; then
continue
fi
echo "cp '${models_path}'/'${model_name}'.ms' ${ms_models_path}'/'${model_name}'.ms" >> "${run_benchmark_log_file}"
cp $models_path/${model_name}.ms ${ms_models_path}/${model_name}.ms
done < ${models_fp16_config}
# Convert tflite models:
while read line; do
model_name=${line}
@ -483,6 +472,16 @@ while read line; do
./converter_lite --fmk=TFLITE --modelFile=${models_path}/${model_name} --outputFile=${ms_models_path}/${model_name} --quantType=AwareTraining || Convert_status=$?
done < ${models_tflite_awaretraining_config}
# Copy fp16 ms models:
while read line; do
model_name=${line%.*}
if [[ $model_name == \#* ]]; then
continue
fi
echo 'cp '${ms_models_path}'/'${model_name}'.ms' ${ms_models_path}'/'${model_name}'.fp16.ms'
cp ${ms_models_path}/${model_name}.ms ${ms_models_path}/${model_name}.fp16.ms
done < ${models_fp16_config}
# Check all result and return value
if [[ ${Convert_status} = 0 ]];then
echo "convert is ended"