Adel-Moumen commited on
Commit
70e4086
1 Parent(s): f3a0147

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +2 -4
hyperparams.yaml CHANGED
@@ -41,7 +41,7 @@ enc: !new:speechbrain.lobes.models.VanillaNN.VanillaNN
41
  dnn_blocks: !ref <dnn_layers>
42
  dnn_neurons: !ref <dnn_neurons>
43
 
44
- wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
45
  source: !ref <wav2vec2_hub>
46
  output_norm: True
47
  freeze: True
@@ -90,15 +90,13 @@ tokenizer: !new:sentencepiece.SentencePieceProcessor
90
  encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
91
  wav2vec2: !ref <wav2vec2>
92
  enc: !ref <enc>
93
-
94
  decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
95
  embedding: !ref <emb>
96
  decoder: !ref <dec>
97
  linear: !ref <seq_lin>
98
- ctc_linear: !ref <ctc_lin>
99
  bos_index: !ref <bos_index>
100
  eos_index: !ref <eos_index>
101
- blank_index: !ref <blank_index>
102
  min_decode_ratio: !ref <min_decode_ratio>
103
  max_decode_ratio: !ref <max_decode_ratio>
104
  beam_size: !ref <beam_size>
41
  dnn_blocks: !ref <dnn_layers>
42
  dnn_neurons: !ref <dnn_neurons>
43
 
44
+ wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
45
  source: !ref <wav2vec2_hub>
46
  output_norm: True
47
  freeze: True
90
  encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
91
  wav2vec2: !ref <wav2vec2>
92
  enc: !ref <enc>
93
+
94
  decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
95
  embedding: !ref <emb>
96
  decoder: !ref <dec>
97
  linear: !ref <seq_lin>
 
98
  bos_index: !ref <bos_index>
99
  eos_index: !ref <eos_index>
 
100
  min_decode_ratio: !ref <min_decode_ratio>
101
  max_decode_ratio: !ref <max_decode_ratio>
102
  beam_size: !ref <beam_size>