Update prepare.sh

k2-fsa · Oct 30, 2024 · 55fea00 · 55fea00
1 parent 18c1b2f
commit 55fea00
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/egs/librilight/SSL/prepare.sh b/egs/librilight/SSL/prepare.sh
@@ -86,15 +86,15 @@ if [ $stage -le 4 ] && [ $stop_stage -ge 4 ]; then
     wget https://dl.fbaipublicfiles.com/hubert/hubert_base_ls960_L9_km500.bin -P download
   fi
   if [ ! -e data/kmeans/.extract_small.done ]; then
-    ./local/discretize_with_s3tokenizer.py --subset small
+    ./local/discretize_with_kmeans.py --subset small
     touch data/kmeans/.extract_small.done
   fi
   if [ ! -e data/kmeans/.extract_medium.done ]; then
-    ./local/discretize_with_s3tokenizer.py --subset medium
+    ./local/discretize_with_kmeans.py --subset medium
     touch data/kmeans/.extract_medium.done
   fi
   if [ ! -e data/kmeans/.extract_large.done ]; then
-    ./local/discretize_with_s3tokenizer.py --subset large
+    ./local/discretize_with_kmeans.py --subset large
     touch data/kmeans/.extract_large.done
   fi
 fi