GaelleLaperriere commited on
Commit
7a1854e
1 Parent(s): f6c0dcc

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +4 -1
hyperparams.yaml CHANGED
@@ -49,6 +49,8 @@ ctc_cost: !name:speechbrain.nnet.losses.ctc_loss
49
  model: !new:torch.nn.ModuleList
50
  - [!ref <enc>, !ref <output_lin>]
51
 
 
 
52
  encoder:
53
  wav2vec2: !ref <wav2vec2>
54
  enc: !ref <enc>
@@ -63,4 +65,5 @@ modules:
63
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
64
  loadables:
65
  wav2vec2: !ref <wav2vec2>
66
- model: !ref <model>
 
 
49
  model: !new:torch.nn.ModuleList
50
  - [!ref <enc>, !ref <output_lin>]
51
 
52
+ tokenizer: !new:speechbrain.dataio.encoder.CTCTextEncoder
53
+
54
  encoder:
55
  wav2vec2: !ref <wav2vec2>
56
  enc: !ref <enc>
 
65
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
66
  loadables:
67
  wav2vec2: !ref <wav2vec2>
68
+ model: !ref <model>
69
+ tokenizer: !ref <tokenizer>