From 3fbfd3e502c47fb9c728a26a938518f92cb18347 Mon Sep 17 00:00:00 2001 From: tianhao zhang <15600919271@163.com> Date: Sat, 19 Nov 2022 08:43:38 +0000 Subject: [PATCH] wav2vec2 demo update: support different optimizer and lr_schedular, align mdoel, update input type, test=asr --- examples/librispeech/asr3/RESULTS.md | 2 +- examples/librispeech/asr3/conf/wav2vec2ASR.yaml | 2 +- paddlespeech/resource/pretrained_models.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/librispeech/asr3/RESULTS.md b/examples/librispeech/asr3/RESULTS.md index f96769d86..27a87e137 100644 --- a/examples/librispeech/asr3/RESULTS.md +++ b/examples/librispeech/asr3/RESULTS.md @@ -5,4 +5,4 @@ train: Epoch 1, 1*V100-32G, batchsize: 6 | Model | Params | Config | Augmentation| Test set | Decode method | WER | | --- | --- | --- | --- | --- | --- | --- | -| wav2vec2ASR | 302.86 M | conf/wav2vec2ASR.yaml | spec_aug | test-clean | greedy search | 0.019 | +| wav2vec2ASR | 302.86 M | conf/wav2vec2ASR.yaml | spec_aug | test-clean | greedy search | 0.018906 | diff --git a/examples/librispeech/asr3/conf/wav2vec2ASR.yaml b/examples/librispeech/asr3/conf/wav2vec2ASR.yaml index 9ca535217..1ce2d94db 100644 --- a/examples/librispeech/asr3/conf/wav2vec2ASR.yaml +++ b/examples/librispeech/asr3/conf/wav2vec2ASR.yaml @@ -108,7 +108,7 @@ audio_augment: # for raw audio ########################################### n_epoch: 1 accum_grad: 1 -global_grad_clip: 3.0 +global_grad_clip: 5.0 model_optim: adadelta model_optim_conf: lr: 0.9 diff --git a/paddlespeech/resource/pretrained_models.py b/paddlespeech/resource/pretrained_models.py index a36c813da..bdab91678 100644 --- a/paddlespeech/resource/pretrained_models.py +++ b/paddlespeech/resource/pretrained_models.py @@ -59,7 +59,7 @@ ssl_dynamic_pretrained_models = { 'url': 'https://paddlespeech.bj.bcebos.com/s2t/librispeech/asr3/wav2vec2ASR-large-960h-librispeech_ckpt_1.3.1.model.tar.gz', 'md5': - '71ee652cdae60fc289a467291017b51c', + 'cbe28d6c78f3dd2e189968402381f454', 'cfg_path': 'model.yaml', 'ckpt_path':