Adel-Moumen
commited on
Update hyperparams.yaml
Browse files- hyperparams.yaml +2 -1
hyperparams.yaml
CHANGED
@@ -48,7 +48,7 @@ enc: !new:speechbrain.nnet.containers.Sequential
|
|
48 |
bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
|
49 |
activation3: !new:torch.nn.LeakyReLU
|
50 |
|
51 |
-
wav2vec2: !new:speechbrain.lobes.models.
|
52 |
source: !ref <wav2vec2_hub>
|
53 |
output_norm: True
|
54 |
freeze: !ref <freeze_wav2vec>
|
@@ -73,6 +73,7 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
|
73 |
wav2vec2: !ref <wav2vec2>
|
74 |
enc: !ref <enc>
|
75 |
ctc_lin: !ref <ctc_lin>
|
|
|
76 |
|
77 |
modules:
|
78 |
encoder: !ref <encoder>
|
|
|
48 |
bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
|
49 |
activation3: !new:torch.nn.LeakyReLU
|
50 |
|
51 |
+
wav2vec2: !new:speechbrain.lobes.models.huggingface_transformers.wav2vec2.Wav2Vec2
|
52 |
source: !ref <wav2vec2_hub>
|
53 |
output_norm: True
|
54 |
freeze: !ref <freeze_wav2vec>
|
|
|
73 |
wav2vec2: !ref <wav2vec2>
|
74 |
enc: !ref <enc>
|
75 |
ctc_lin: !ref <ctc_lin>
|
76 |
+
log_softmax: !ref <log_softmax>
|
77 |
|
78 |
modules:
|
79 |
encoder: !ref <encoder>
|