diff --git a/examples/librispeech/asr5/conf/wavlmASR.yaml b/examples/librispeech/asr5/conf/wavlmASR.yaml index 5df2bc629..25f9643e9 100644 --- a/examples/librispeech/asr5/conf/wavlmASR.yaml +++ b/examples/librispeech/asr5/conf/wavlmASR.yaml @@ -16,7 +16,7 @@ ctc: enc_n_units: 768 blank_id: 0 dropout_rate: 0.0 -wavlm_params_path: "/home/ubuntu/Documents/Github/wavlm_paddle/wavlm-paddle-ft.pth" +wavlm_params_path: exp/wavlm/wavlm-base-plus.pdparams task_cfg: diff --git a/examples/librispeech/asr5/local/data.sh b/examples/librispeech/asr5/local/data.sh index d243f297b..8e69dd769 100644 --- a/examples/librispeech/asr5/local/data.sh +++ b/examples/librispeech/asr5/local/data.sh @@ -101,10 +101,10 @@ fi echo "LibriSpeech Data preparation done." -# if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then -# mkdir -p exp/hubert -# echo "Pretrained hubert model download" -# wget -P exp/hubert https://paddlespeech.bj.bcebos.com/hubert/hubert-large-lv60.pdparams -# fi +if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then + mkdir -p exp/wavlm + echo "Pretrained wavlm model download" + wget -P exp/wavlm https://paddlespeech.bj.bcebos.com/wavlm/wavlm-base-plus.pdparams +fi exit 0 \ No newline at end of file