游雁 2 роки тому
батько
коміт
adf3237662
2 змінених файлів з 15 додано та 5 видалено
  1. 11 4
      egs/aishell/transformer/utils/apply_cmvn.sh
  2. 4 1
      funasr/__init__.py

+ 11 - 4
egs/aishell/transformer/utils/apply_cmvn.sh

@@ -17,10 +17,17 @@ output_dir=$4
 dump_dir=${output_dir}/ark; mkdir -p ${dump_dir}
 mkdir -p ${logdir}
 
-$cmd JOB=1:$nj $logdir/apply_cmvn.JOB.log \
-    python utils/apply_cmvn.py -a $fbankdir/ark/feats.JOB.ark \
-        -c $cmvn_file -i JOB -o ${dump_dir} \
-        || exit 1;
+#$cmd JOB=1:$nj $logdir/apply_cmvn.JOB.log \
+#    python utils/apply_cmvn.py -a $fbankdir/ark/feats.JOB.ark \
+#        -c $cmvn_file -i JOB -o ${dump_dir} \
+#        || exit 1;
+
+for JOB in `seq 1 $nj`;do
+  {
+      python utils/apply_cmvn.py -a $fbankdir/ark/feats.${JOB}.ark \
+    -c $cmvn_file -i ${JOB} -o ${dump_dir} || exit 1;
+  } &> $logdir/apply_cmvn.${JOB}.log &
+done
 
 for n in $(seq $nj); do
     cat ${dump_dir}/feats.$n.scp || exit 1

+ 4 - 1
funasr/__init__.py

@@ -129,4 +129,7 @@ def infer(task_name: str = "asr",
         
         return inference_pipeline(data_path_and_name_and_type, raw_inputs=raw_inputs, **kwargs)
     
-    return _infer_fn
+    return _infer_fn
+
+if __name__ == '__main__':
+    pass