提交 7dd18c0d 编写于 作者: M meng_chunyang

update the copy rules for fp16 models

上级 34cdab58
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
......@@ -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"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册