speechbrain
/

asr-wav2vec2-ctc-aishell

Model card Files Files and versions Community

Adel-Moumen commited on Feb 19

Commit

9531604

•

1 Parent(s): 0526b7c

Update hyperparams.yaml

Files changed (1) hide show

hyperparams.yaml +2 -1

hyperparams.yaml CHANGED Viewed

@@ -47,7 +47,7 @@ enc: !new:speechbrain.nnet.containers.Sequential
    bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
    activation3: !new:torch.nn.LeakyReLU
-wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
    source: !ref <wav2vec2_hub>
    output_norm: True
    freeze: !ref <freeze_wav2vec>
@@ -70,6 +70,7 @@ modules:
    wav2vec2: !ref <wav2vec2>
    enc: !ref <enc>
    ctc_lin: !ref <ctc_lin>
 pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
    loadables:

    bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
    activation3: !new:torch.nn.LeakyReLU
+wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
    source: !ref <wav2vec2_hub>
    output_norm: True
    freeze: !ref <freeze_wav2vec>
    wav2vec2: !ref <wav2vec2>
    enc: !ref <enc>
    ctc_lin: !ref <ctc_lin>
+   log_softmax: !ref <log_softmax>
 pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
    loadables: