嘉渊 2 سال پیش
والد
کامیت
eb3948b480
2فایلهای تغییر یافته به همراه4 افزوده شده و 2 حذف شده
  1. 2 1
      egs/aishell/paraformerbert/run.sh
  2. 2 1
      egs/aishell2/paraformerbert/run.sh

+ 2 - 1
egs/aishell/paraformerbert/run.sh

@@ -115,7 +115,8 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
         echo "extract embeddings..."
         local/extract_embeds.sh \
             --bert_model_name ${bert_model_name} \
-            --raw_dataset_path ${feats_dir}
+            --raw_dataset_path ${feats_dir} \
+            --nj $nj
     fi
     mkdir -p ${exp_dir}/exp/${model_dir}
     mkdir -p ${exp_dir}/exp/${model_dir}/log

+ 2 - 1
egs/aishell2/paraformerbert/run.sh

@@ -116,7 +116,8 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
         echo "extract embeddings..."
         local/extract_embeds.sh \
             --bert_model_name ${bert_model_name} \
-            --raw_dataset_path ${feats_dir}
+            --raw_dataset_path ${feats_dir} \
+            --nj $nj
     fi
     mkdir -p ${exp_dir}/exp/${model_dir}
     mkdir -p ${exp_dir}/exp/${model_dir}/log