Update hyperparams.yaml
Browse files- hyperparams.yaml +1 -1
hyperparams.yaml
CHANGED
@@ -19,7 +19,7 @@ n_mels: 80
|
|
19 |
####################### Model parameters ###########################
|
20 |
# Transformer
|
21 |
d_model: 256
|
22 |
-
nhead:
|
23 |
num_encoder_layers: 12
|
24 |
num_decoder_layers: 6
|
25 |
d_ffn: 2048
|
|
|
19 |
####################### Model parameters ###########################
|
20 |
# Transformer
|
21 |
d_model: 256
|
22 |
+
nhead: 4
|
23 |
num_encoder_layers: 12
|
24 |
num_decoder_layers: 6
|
25 |
d_ffn: 2048
|