speechbrain
/

asr-wav2vec2-commonvoice-14-pt

Automatic Speech Recognition

Model card Files Files and versions Community

Adel-Moumen commited on Feb 19

Commit

05e73fb

•

1 Parent(s): 5cc0eae

Update hyperparams.yaml

Files changed (1) hide show

hyperparams.yaml +2 -1

hyperparams.yaml CHANGED Viewed

@@ -51,7 +51,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
@@ -76,6 +76,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>