fix the fixed num_splits (#1772)

k2-fsa · Oct 16, 2024 · f84270c · f84270c
1 parent 2653df5
commit f84270c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/egs/wenetspeech/ASR/prepare.sh b/egs/wenetspeech/ASR/prepare.sh
@@ -161,23 +161,23 @@ fi
 if [ $stage -le 11 ] && [ $stop_stage -ge 11 ]; then
   log "Stage 11: Combine features for S"
   if [ ! -f data/fbank/cuts_S.jsonl.gz ]; then
-    pieces=$(find data/fbank/S_split_1000 -name "cuts_S.*.jsonl.gz")
+    pieces=$(find data/fbank/S_split_${num_splits} -name "cuts_S.*.jsonl.gz")
     lhotse combine $pieces data/fbank/cuts_S.jsonl.gz
   fi
 fi
 
 if [ $stage -le 12 ] && [ $stop_stage -ge 12 ]; then
   log "Stage 12: Combine features for M"
   if [ ! -f data/fbank/cuts_M.jsonl.gz ]; then
-    pieces=$(find data/fbank/M_split_1000 -name "cuts_M.*.jsonl.gz")
+    pieces=$(find data/fbank/M_split_${num_splits} -name "cuts_M.*.jsonl.gz")
     lhotse combine $pieces data/fbank/cuts_M.jsonl.gz
   fi
 fi
 
 if [ $stage -le 13 ] && [ $stop_stage -ge 13 ]; then
   log "Stage 13: Combine features for L"
   if [ ! -f data/fbank/cuts_L.jsonl.gz ]; then
-    pieces=$(find data/fbank/L_split_1000 -name "cuts_L.*.jsonl.gz")
+    pieces=$(find data/fbank/L_split_${num_splits} -name "cuts_L.*.jsonl.gz")
     lhotse combine $pieces data/fbank/cuts_L.jsonl.gz
   fi
 fi