Commit
·
8fb47b1
1
Parent(s):
b8e59f8
Update hyperparams
Browse files- hyperparams.yaml +2 -5
hyperparams.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
# ################################
|
| 2 |
# Model: wav2vec2 + DNN + CTC/Attention
|
| 3 |
# Augmentation: SpecAugment
|
| 4 |
-
# Authors:
|
| 5 |
# ################################
|
| 6 |
|
| 7 |
sample_rate: 16000
|
|
@@ -11,8 +11,6 @@ wav2vec2_hub: dragonSwing/wav2vec2-base-vn-270h
|
|
| 11 |
activation: !name:torch.nn.LeakyReLU
|
| 12 |
dnn_layers: 2
|
| 13 |
dnn_neurons: 768
|
| 14 |
-
emb_size: 128
|
| 15 |
-
dec_neurons: 768
|
| 16 |
dropout_prob: 0.1
|
| 17 |
|
| 18 |
# Outputs
|
|
@@ -36,8 +34,7 @@ wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
|
|
| 36 |
source: !ref <wav2vec2_hub>
|
| 37 |
output_norm: True
|
| 38 |
freeze: True
|
| 39 |
-
|
| 40 |
-
save_path: null
|
| 41 |
|
| 42 |
lm_head: !new:torch.nn.Linear
|
| 43 |
in_features: !ref <dnn_neurons>
|
|
|
|
| 1 |
# ################################
|
| 2 |
# Model: wav2vec2 + DNN + CTC/Attention
|
| 3 |
# Augmentation: SpecAugment
|
| 4 |
+
# Authors: Le Do Thanh Binh 2021
|
| 5 |
# ################################
|
| 6 |
|
| 7 |
sample_rate: 16000
|
|
|
|
| 11 |
activation: !name:torch.nn.LeakyReLU
|
| 12 |
dnn_layers: 2
|
| 13 |
dnn_neurons: 768
|
|
|
|
|
|
|
| 14 |
dropout_prob: 0.1
|
| 15 |
|
| 16 |
# Outputs
|
|
|
|
| 34 |
source: !ref <wav2vec2_hub>
|
| 35 |
output_norm: True
|
| 36 |
freeze: True
|
| 37 |
+
save_path: model_checkpoints
|
|
|
|
| 38 |
|
| 39 |
lm_head: !new:torch.nn.Linear
|
| 40 |
in_features: !ref <dnn_neurons>
|