Adel-Moumen commited on
Commit
269c16b
1 Parent(s): a83e454

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +2 -1
hyperparams.yaml CHANGED
@@ -58,7 +58,7 @@ enc: !new:speechbrain.nnet.containers.Sequential
58
  bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
59
  activation3: !new:torch.nn.LeakyReLU
60
 
61
- wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
62
  source: !ref <wav2vec2_hub>
63
  output_norm: True
64
  freeze: True
@@ -83,6 +83,7 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
83
  wav2vec2: !ref <wav2vec2>
84
  enc: !ref <enc>
85
  ctc_lin: !ref <ctc_lin>
 
86
 
87
  decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
88
  blank_id: !ref <blank_index>
 
58
  bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
59
  activation3: !new:torch.nn.LeakyReLU
60
 
61
+ wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
62
  source: !ref <wav2vec2_hub>
63
  output_norm: True
64
  freeze: True
 
83
  wav2vec2: !ref <wav2vec2>
84
  enc: !ref <enc>
85
  ctc_lin: !ref <ctc_lin>
86
+ log_softmax: !ref <log_softmax>
87
 
88
  decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
89
  blank_id: !ref <blank_index>