Merge pull request #5731 from Emrys365/tse_recipe

Update TSE recipe egs2/librimix/tse1
espnet · Apr 9, 2024 · f6f011d · f6f011d
2 parents 58d348a + 038de87
commit f6f011d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/egs2/librimix/tse1/conf/tuning/train_enh_tse_td_speakerbeam_16k.yaml b/egs2/librimix/tse1/conf/tuning/train_enh_tse_td_speakerbeam_16k.yaml
@@ -38,10 +38,13 @@ model_conf:
     num_spk: 2
     share_encoder: true
 
+# preprocessor-related arguments
+# see TSEPreprocessor in https://github.com/espnet/espnet/blob/master/espnet2/train/preprocessor.py
 train_spk2enroll: data/train-100/spk2enroll.json
 enroll_segment: 48000
 load_spk_embedding: false
-load_all_speakers: false
+load_all_speakers: true
+num_skp: 2
 
 encoder: conv
 encoder_conf:

diff --git a/egs2/librimix/tse1/local/asr_data_prep.sh b/egs2/librimix/tse1/local/asr_data_prep.sh
@@ -0,0 +1 @@
+../../asr1/local/data_prep.sh