DAE-TSE Pretrained Models

KCE (Keyword-guided Cue Encoder)

• File: kce/epoch_149.pt
• Architecture: AEDKWSASRPhone
• Training data: LibriSpeech train-clean-360 + train-other-500
• Validation data: LibriSpeech train-clean-100
• Config: kce/model.yaml, kce/data.yaml

Backbone (DAE-BSRNN)

• File: backbone/models/checkpoint_150.pt
• Architecture: DAE-BSRNN (text-aware BSRNN with TC-ASR encoder)
• Training data: Libri2Mix-100 mix-clean
• Config: backbone/config.yaml

After downloading, update model_args.tse_model.asr_encoder.checkpoint_dir in backbone/config.yaml to the absolute path of kce/epoch_149.pt.