Commit
•
bbb3d6f
1
Parent(s):
a516085
Adding the acoustic model (.nemo) and the architecture (.yaml) to the Repo
Browse files- .gitattributes +1 -0
- QuartzNet_FT15x5_Icelandic.yaml +278 -0
- stt_is_quartznet15x5_ft_ep56_875h.nemo +3 -0
.gitattributes
CHANGED
@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
32 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
33 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
34 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
32 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
33 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
34 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
35 |
+
stt_is_quartznet15x5_ft_ep56_875h.nemo filter=lfs diff=lfs merge=lfs -text
|
QuartzNet_FT15x5_Icelandic.yaml
ADDED
@@ -0,0 +1,278 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
name: &name "QuartzNet15x5"
|
2 |
+
|
3 |
+
model:
|
4 |
+
sample_rate: &sample_rate 16000
|
5 |
+
repeat: &repeat 5
|
6 |
+
dropout: &dropout 0.0
|
7 |
+
separable: &separable true
|
8 |
+
labels: &labels [" ", "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z", "á", "æ", "é", "í", "ð", "ó", "ö", "ú", "ý", "þ"]
|
9 |
+
|
10 |
+
train_ds:
|
11 |
+
manifest_filepath: ???
|
12 |
+
sample_rate: 16000
|
13 |
+
labels: *labels
|
14 |
+
batch_size: 16 ##########################
|
15 |
+
trim_silence: True
|
16 |
+
max_duration: 16.7
|
17 |
+
shuffle: True
|
18 |
+
num_workers: 8
|
19 |
+
pin_memory: true
|
20 |
+
# tarred datasets
|
21 |
+
is_tarred: false
|
22 |
+
tarred_audio_filepaths: null
|
23 |
+
shuffle_n: 2048
|
24 |
+
# bucketing params
|
25 |
+
bucketing_strategy: "synced_randomized"
|
26 |
+
bucketing_batch_size: null
|
27 |
+
|
28 |
+
validation_ds:
|
29 |
+
manifest_filepath: ???
|
30 |
+
sample_rate: 16000
|
31 |
+
labels: *labels
|
32 |
+
batch_size: 16 ##########################
|
33 |
+
shuffle: False
|
34 |
+
num_workers: 8
|
35 |
+
pin_memory: true
|
36 |
+
|
37 |
+
preprocessor:
|
38 |
+
_target_: nemo.collections.asr.modules.AudioToMelSpectrogramPreprocessor
|
39 |
+
normalize: "per_feature"
|
40 |
+
window_size: 0.02
|
41 |
+
sample_rate: *sample_rate
|
42 |
+
window_stride: 0.01
|
43 |
+
window: "hann"
|
44 |
+
features: &n_mels 64
|
45 |
+
n_fft: 512
|
46 |
+
frame_splicing: 1
|
47 |
+
dither: 1.0e-05
|
48 |
+
|
49 |
+
spec_augment:
|
50 |
+
_target_: nemo.collections.asr.modules.SpectrogramAugmentation
|
51 |
+
rect_freq: 50
|
52 |
+
rect_masks: 5
|
53 |
+
rect_time: 120
|
54 |
+
|
55 |
+
encoder:
|
56 |
+
_target_: nemo.collections.asr.modules.ConvASREncoder
|
57 |
+
feat_in: *n_mels
|
58 |
+
activation: relu
|
59 |
+
conv_mask: true
|
60 |
+
|
61 |
+
jasper:
|
62 |
+
#1
|
63 |
+
- dilation: [1]
|
64 |
+
dropout: *dropout
|
65 |
+
filters: 256
|
66 |
+
kernel: [33]
|
67 |
+
repeat: 1
|
68 |
+
residual: false
|
69 |
+
separable: *separable
|
70 |
+
stride: [2]
|
71 |
+
#2
|
72 |
+
- dilation: [1]
|
73 |
+
dropout: *dropout
|
74 |
+
filters: 256
|
75 |
+
kernel: [33]
|
76 |
+
repeat: *repeat
|
77 |
+
residual: true
|
78 |
+
separable: *separable
|
79 |
+
stride: [1]
|
80 |
+
#3
|
81 |
+
- dilation: [1]
|
82 |
+
dropout: *dropout
|
83 |
+
filters: 256
|
84 |
+
kernel: [33]
|
85 |
+
repeat: *repeat
|
86 |
+
residual: true
|
87 |
+
separable: *separable
|
88 |
+
stride: [1]
|
89 |
+
#4
|
90 |
+
- dilation: [1]
|
91 |
+
dropout: *dropout
|
92 |
+
filters: 256
|
93 |
+
kernel: [33]
|
94 |
+
repeat: *repeat
|
95 |
+
residual: true
|
96 |
+
separable: *separable
|
97 |
+
stride: [1]
|
98 |
+
#5
|
99 |
+
- dilation: [1]
|
100 |
+
dropout: *dropout
|
101 |
+
filters: 256
|
102 |
+
kernel: [39]
|
103 |
+
repeat: *repeat
|
104 |
+
residual: true
|
105 |
+
separable: *separable
|
106 |
+
stride: [1]
|
107 |
+
#6
|
108 |
+
- dilation: [1]
|
109 |
+
dropout: *dropout
|
110 |
+
filters: 256
|
111 |
+
kernel: [39]
|
112 |
+
repeat: *repeat
|
113 |
+
residual: true
|
114 |
+
separable: *separable
|
115 |
+
stride: [1]
|
116 |
+
#7
|
117 |
+
- dilation: [1]
|
118 |
+
dropout: *dropout
|
119 |
+
filters: 256
|
120 |
+
kernel: [39]
|
121 |
+
repeat: *repeat
|
122 |
+
residual: true
|
123 |
+
separable: *separable
|
124 |
+
stride: [1]
|
125 |
+
#8
|
126 |
+
- dilation: [1]
|
127 |
+
dropout: *dropout
|
128 |
+
filters: 512
|
129 |
+
kernel: [51]
|
130 |
+
repeat: *repeat
|
131 |
+
residual: true
|
132 |
+
separable: *separable
|
133 |
+
stride: [1]
|
134 |
+
#9
|
135 |
+
- dilation: [1]
|
136 |
+
dropout: *dropout
|
137 |
+
filters: 512
|
138 |
+
kernel: [51]
|
139 |
+
repeat: *repeat
|
140 |
+
residual: true
|
141 |
+
separable: *separable
|
142 |
+
stride: [1]
|
143 |
+
#10
|
144 |
+
- dilation: [1]
|
145 |
+
dropout: *dropout
|
146 |
+
filters: 512
|
147 |
+
kernel: [51]
|
148 |
+
repeat: *repeat
|
149 |
+
residual: true
|
150 |
+
separable: *separable
|
151 |
+
stride: [1]
|
152 |
+
#11
|
153 |
+
- dilation: [1]
|
154 |
+
dropout: *dropout
|
155 |
+
filters: 512
|
156 |
+
kernel: [63]
|
157 |
+
repeat: *repeat
|
158 |
+
residual: true
|
159 |
+
separable: *separable
|
160 |
+
stride: [1]
|
161 |
+
#12
|
162 |
+
- dilation: [1]
|
163 |
+
dropout: *dropout
|
164 |
+
filters: 512
|
165 |
+
kernel: [63]
|
166 |
+
repeat: *repeat
|
167 |
+
residual: true
|
168 |
+
separable: *separable
|
169 |
+
stride: [1]
|
170 |
+
#13
|
171 |
+
- dilation: [1]
|
172 |
+
dropout: *dropout
|
173 |
+
filters: 512
|
174 |
+
kernel: [63]
|
175 |
+
repeat: *repeat
|
176 |
+
residual: true
|
177 |
+
separable: *separable
|
178 |
+
stride: [1]
|
179 |
+
#14
|
180 |
+
- dilation: [1]
|
181 |
+
dropout: *dropout
|
182 |
+
filters: 512
|
183 |
+
kernel: [75]
|
184 |
+
repeat: *repeat
|
185 |
+
residual: true
|
186 |
+
separable: *separable
|
187 |
+
stride: [1]
|
188 |
+
#15
|
189 |
+
- dilation: [1]
|
190 |
+
dropout: *dropout
|
191 |
+
filters: 512
|
192 |
+
kernel: [75]
|
193 |
+
repeat: *repeat
|
194 |
+
residual: true
|
195 |
+
separable: *separable
|
196 |
+
stride: [1]
|
197 |
+
#16
|
198 |
+
- dilation: [1]
|
199 |
+
dropout: *dropout
|
200 |
+
filters: 512
|
201 |
+
kernel: [75]
|
202 |
+
repeat: *repeat
|
203 |
+
residual: true
|
204 |
+
separable: *separable
|
205 |
+
stride: [1]
|
206 |
+
#17
|
207 |
+
- dilation: [2]
|
208 |
+
dropout: *dropout
|
209 |
+
filters: 512
|
210 |
+
kernel: [87]
|
211 |
+
repeat: 1
|
212 |
+
residual: false
|
213 |
+
separable: *separable
|
214 |
+
stride: [1]
|
215 |
+
#18
|
216 |
+
- dilation: [1]
|
217 |
+
dropout: *dropout
|
218 |
+
filters: &enc_filters 1024
|
219 |
+
kernel: [1]
|
220 |
+
repeat: 1
|
221 |
+
residual: false
|
222 |
+
stride: [1]
|
223 |
+
|
224 |
+
decoder:
|
225 |
+
_target_: nemo.collections.asr.modules.ConvASRDecoder
|
226 |
+
feat_in: *enc_filters
|
227 |
+
num_classes: 37
|
228 |
+
vocabulary: *labels
|
229 |
+
|
230 |
+
optim:
|
231 |
+
name: novograd
|
232 |
+
# _target_: nemo.core.optim.optimizers.Novograd
|
233 |
+
lr: 0.0012
|
234 |
+
# optimizer arguments
|
235 |
+
betas: [0.95, 0.25]
|
236 |
+
weight_decay: 0.001
|
237 |
+
|
238 |
+
# scheduler setup
|
239 |
+
sched:
|
240 |
+
name: CosineAnnealing
|
241 |
+
|
242 |
+
# pytorch lightning args
|
243 |
+
# monitor: val_loss
|
244 |
+
# reduce_on_plateau: false
|
245 |
+
|
246 |
+
# Scheduler params
|
247 |
+
warmup_steps: null
|
248 |
+
warmup_ratio: null
|
249 |
+
min_lr: 0.0
|
250 |
+
last_epoch: -1
|
251 |
+
|
252 |
+
trainer:
|
253 |
+
devices: 1 # number of gpus
|
254 |
+
max_epochs: 5
|
255 |
+
max_steps: -1 # computed at runtime if not set
|
256 |
+
num_nodes: 1
|
257 |
+
accelerator: gpu
|
258 |
+
strategy: ddp
|
259 |
+
accumulate_grad_batches: 1
|
260 |
+
enable_checkpointing: False # Provided by exp_manager
|
261 |
+
logger: False # Provided by exp_manager
|
262 |
+
log_every_n_steps: 1 # Interval of logging.
|
263 |
+
val_check_interval: 1.0 # Set to 0.25 to check 4 times per epoch, or an int for number of iterations
|
264 |
+
benchmark: false # needs to be false for models with variable-length speech input as it slows down training
|
265 |
+
|
266 |
+
exp_manager:
|
267 |
+
exp_dir: null
|
268 |
+
name: *name
|
269 |
+
create_tensorboard_logger: True
|
270 |
+
create_checkpoint_callback: True
|
271 |
+
checkpoint_callback_params:
|
272 |
+
monitor: "val_wer"
|
273 |
+
mode: "min"
|
274 |
+
create_wandb_logger: False
|
275 |
+
wandb_logger_kwargs:
|
276 |
+
name: null
|
277 |
+
project: null
|
278 |
+
|
stt_is_quartznet15x5_ft_ep56_875h.nemo
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:470116b575b5e9a84f5bbfad7755c5b7e13b364bb99e2dc15137920669353a29
|
3 |
+
size 76359680
|