update

examples/mpnet_aishell/run.sh

@@ -3,16 +3,16 @@
 : <<'END'
 
 
-sh run.sh --stage 2 --stop_stage 2 --system_version windows --file_folder_name file_dir \
+sh run.sh --stage 2 --stop_stage 2 --system_version windows --file_folder_name file_dir --final_model_name mpnet-aishell-20250224 \
 --noise_dir "E:/Users/tianx/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "E:/programmer/asr_datasets/aishell/data_aishell/wav/train"
 
 
-sh run.sh --stage 3 --stop_stage 3 --system_version centos --file_folder_name file_dir \
+sh run.sh --stage 3 --stop_stage 3 --system_version centos --file_folder_name file_dir --final_model_name mpnet-aishell-20250224 \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
 
-sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir \
+sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name mpnet-aishell-20250224 \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
 
@@ -130,31 +130,12 @@ fi
 
 
 if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
-  $verbose && echo "stage 4: export model"
-  cd "${work_dir}" || exit 1
-  python3 step_5_export_models.py \
-  --vocabulary_dir "${vocabulary_dir}" \
-  --model_dir "${file_dir}/best" \
-  --serialization_dir "${file_dir}" \
-
-fi
-
-
-if [ ${stage} -le 5 ] && [ ${stop_stage} -ge 5 ]; then
-  $verbose && echo "stage 5: collect files"
+  $verbose && echo "stage 4: collect files"
   cd "${work_dir}" || exit 1
 
   mkdir -p ${final_model_dir}
 
   cp "${file_dir}/best"/* "${final_model_dir}"
-  cp -r "${file_dir}/vocabulary" "${final_model_dir}"
-
-  cp "${file_dir}/evaluation.xlsx" "${final_model_dir}/evaluation.xlsx"
-
-  cp "${file_dir}/trace_model.zip" "${final_model_dir}/trace_model.zip"
-  cp "${file_dir}/trace_quant_model.zip" "${final_model_dir}/trace_quant_model.zip"
-  cp "${file_dir}/script_model.zip" "${final_model_dir}/script_model.zip"
-  cp "${file_dir}/script_quant_model.zip" "${final_model_dir}/script_quant_model.zip"
 
   cd "${final_model_dir}/.." || exit 1;
 
@@ -169,8 +150,8 @@ if [ ${stage} -le 5 ] && [ ${stop_stage} -ge 5 ]; then
 fi
 
 
-if [ ${stage} -le 6 ] && [ ${stop_stage} -ge 6 ]; then
-  $verbose && echo "stage 6: clear file_dir"
+if [ ${stage} -le 5 ] && [ ${stop_stage} -ge 5 ]; then
+  $verbose && echo "stage 5: clear file_dir"
   cd "${work_dir}" || exit 1
 
   rm -rf "${file_dir}";