Update hyperparams.yaml
Browse files- hyperparams.yaml +3 -3
hyperparams.yaml
CHANGED
|
@@ -49,10 +49,10 @@ lm_model: !new:speechbrain.lobes.models.RNNLM.RNNLM
|
|
| 49 |
embedding_dim: !ref <emb_size>
|
| 50 |
activation: !name:torch.nn.LeakyReLU
|
| 51 |
dropout: 0.0
|
| 52 |
-
rnn_layers:
|
| 53 |
rnn_neurons: 2048
|
| 54 |
-
dnn_blocks:
|
| 55 |
-
dnn_neurons:
|
| 56 |
return_hidden: True # For inference
|
| 57 |
|
| 58 |
# Wav2vec2 encoder
|
|
|
|
| 49 |
embedding_dim: !ref <emb_size>
|
| 50 |
activation: !name:torch.nn.LeakyReLU
|
| 51 |
dropout: 0.0
|
| 52 |
+
rnn_layers: 3
|
| 53 |
rnn_neurons: 2048
|
| 54 |
+
dnn_blocks: 2
|
| 55 |
+
dnn_neurons: 1024
|
| 56 |
return_hidden: True # For inference
|
| 57 |
|
| 58 |
# Wav2vec2 encoder
|