speechbrain
/

asr-wav2vec2-commonvoice-de

Automatic Speech Recognition

Model card Files Files and versions Community

Adel-Moumen commited on Feb 20, 2024

Commit

02e1f30

·

verified ·

1 Parent(s): fbcc994

Update hyperparams.yaml

Files changed (1) hide show

hyperparams.yaml +2 -1

hyperparams.yaml CHANGED Viewed

@@ -52,7 +52,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: True
@@ -77,6 +77,7 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
     wav2vec2: !ref <wav2vec2>
     enc: !ref <enc>
     ctc_lin: !ref <ctc_lin>
 modules:
   encoder: !ref <encoder>

   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: True
     wav2vec2: !ref <wav2vec2>
     enc: !ref <enc>
     ctc_lin: !ref <ctc_lin>
+    log_softmax: !ref <log_softmax>
 modules:
   encoder: !ref <encoder>