Adel-Moumen commited on
Commit
2433149
1 Parent(s): 4a9e48d

Update hyperparams.yaml

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