speechbrain
/

SLU-direct-SLURP-hubert-enc

Spoken language understanding

Model card Files Files and versions Community

Adel-Moumen commited on Feb 19

Commit

0f76912

•

1 Parent(s): 7665355

Update hyperparams.yaml

Files changed (1) hide show

hyperparams.yaml +1 -2

hyperparams.yaml CHANGED Viewed

@@ -22,7 +22,7 @@ dec_neurons: 512
 dec_attn_dim: 512
 dec_layer: 3
-hubert: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
     source: !ref <wav2vec2_hub>
     output_norm: True
     freeze: True
@@ -60,7 +60,6 @@ beam_searcher: !new:speechbrain.decoders.S2SRNNBeamSearcher
   temperature: 1.25
   using_max_attn_shift: false
   max_attn_shift: 30
-  coverage_penalty: 0.
 model: !new:torch.nn.ModuleList
     - [!ref <output_emb>, !ref <dec>, !ref <seq_lin>]

 dec_attn_dim: 512
 dec_layer: 3
+hubert: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
     source: !ref <wav2vec2_hub>
     output_norm: True
     freeze: True
   temperature: 1.25
   using_max_attn_shift: false
   max_attn_shift: 30
 model: !new:torch.nn.ModuleList
     - [!ref <output_emb>, !ref <dec>, !ref <seq_lin>]