Adel-Moumen commited on
Commit
356e347
1 Parent(s): 9035a95

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +3 -2
hyperparams.yaml CHANGED
@@ -28,7 +28,7 @@ enc: !new:speechbrain.lobes.models.VanillaNN.VanillaNN
28
  dnn_blocks: !ref <dnn_layers>
29
  dnn_neurons: !ref <dnn_neurons>
30
 
31
- wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
32
  source: !ref <wav2vec2_hub>
33
  output_norm: True
34
  freeze: True
@@ -53,6 +53,7 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
53
  wav2vec2: !ref <wav2vec2>
54
  enc: !ref <enc>
55
  ctc_lin: !ref <ctc_lin>
 
56
 
57
  decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
58
  blank_id: !ref <blank_index>
@@ -64,4 +65,4 @@ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
64
  loadables:
65
  wav2vec2: !ref <wav2vec2>
66
  asr: !ref <asr_model>
67
- tokenizer: !ref <tokenizer>
 
28
  dnn_blocks: !ref <dnn_layers>
29
  dnn_neurons: !ref <dnn_neurons>
30
 
31
+ wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.Wav2Vec2
32
  source: !ref <wav2vec2_hub>
33
  output_norm: True
34
  freeze: True
 
53
  wav2vec2: !ref <wav2vec2>
54
  enc: !ref <enc>
55
  ctc_lin: !ref <ctc_lin>
56
+ log_softmax: !ref <log_softmax>
57
 
58
  decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
59
  blank_id: !ref <blank_index>
 
65
  loadables:
66
  wav2vec2: !ref <wav2vec2>
67
  asr: !ref <asr_model>
68
+ tokenizer: !ref <tokenizer>