Vadim commited on
Commit
e484d37
·
1 Parent(s): 9fba329

add QuartzNet15x5

Browse files
README.md CHANGED
@@ -23,6 +23,16 @@ tags:
23
  - _last checkpoint:_ ```quartznet15x5/2023-04-04_02-57-02/QuartzNet15x5--val_loss=1.9543-epoch=34-last.ckpt```
24
  - _Dataset train:_ **1042635** files totalling **1028.14** hours
25
  - _Dataset validation:_ **50640** files totalling **50.71** hours
 
 
 
 
 
 
 
 
 
 
26
 
27
 
28
  ## Jasper10x5
 
23
  - _last checkpoint:_ ```quartznet15x5/2023-04-04_02-57-02/QuartzNet15x5--val_loss=1.9543-epoch=34-last.ckpt```
24
  - _Dataset train:_ **1042635** files totalling **1028.14** hours
25
  - _Dataset validation:_ **50640** files totalling **50.71** hours
26
+ - 2023-02-23
27
+ - _language:_ uk-RU
28
+ - _from_pretrained:_ [stt_en_quartznet15x5](https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/stt_en_quartznet15x5)
29
+ - _Epoch:_ 7
30
+ - _Step:_ 217287
31
+ - _val_loss:_ 1.66200 (best 1.66200)
32
+ - _model:_ ```quartznet15x5/2023-02-23_11-43-23/stt_uk_ru_quartznet15x5.nemo```
33
+ - _last checkpoint:_ ```quartznet15x5/2023-02-23_11-43-23/QuartzNet15x5--val_loss=1.9543-epoch=34-last.ckpt```
34
+ - _Dataset train:_ **1738252** files totalling **1645.74** hours
35
+ - _Dataset validation:_ **11373** files totalling **12.63** hours
36
 
37
 
38
  ## Jasper10x5
quartznet15x5/2023-02-23_11-43-23/QuartzNet15x5--val_loss=1.6620-epoch=7-last.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15c24c27f9168713d36e02df2213d2b0358d07c05fe0dceb013709f01599ec19
3
+ size 152422731
quartznet15x5/2023-02-23_11-43-23/README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ docker:
2
+ ```shell
3
+ docker run --gpus all --ulimit memlock=-1 --ulimit stack=67108864 --shm-size=12gb -p 8888:8888 -p 6006:6006 -it --rm -v ${PWD}/ds:/ds -v ${PWD}/../../datasets:/datasets vadik/nemo-asr-local:1.15
4
+ ```
5
+
6
+ Usage:
7
+ ```python
8
+ import pytorch_lightning as pl
9
+ from nemo.core.config import hydra_runner
10
+ from nemo.utils import logging
11
+ from nemo.utils.exp_manager import exp_manager
12
+ from nemo.collections.asr.models import EncDecCTCModel, configs
13
+ from nemo.utils.config_utils import update_model_config
14
+ import nemo.collections.asr as nemo_asr
15
+ import copy
16
+ from omegaconf import DictConfig, OmegaConf
17
+
18
+
19
+ @hydra_runner(config_path="conf", config_name="stt_uk_ru_quartznet15x5")
20
+ def main(cfg: DictConfig):
21
+ """
22
+ >>> HYDRA_FULL_ERROR=1 python /ds/training_stt_en_quartznet15x5_v4.py
23
+ """
24
+ logging.info(f'Конфігурація Hydra')
25
+
26
+ model = nemo_asr.models.EncDecCTCModel.from_pretrained(
27
+ model_name="stt_en_quartznet15x5"
28
+ ) # EncDecCTCModel
29
+
30
+ model.change_vocabulary(
31
+ new_vocabulary=cfg['labels']
32
+ )
33
+
34
+ new_opt = copy.deepcopy(cfg['model']['optim'])
35
+ # Use the smaller learning rate we set before
36
+ model.setup_optimization(optim_config=DictConfig(new_opt))
37
+
38
+ model.setup_training_data(train_data_config=cfg['model']['train_ds'])
39
+ model.setup_validation_data(val_data_config=cfg['model']['validation_ds'])
40
+
41
+ # And now we can create a PyTorch Lightning trainer and call `fit` again.
42
+ trainer = pl.Trainer(**cfg['trainer'])
43
+
44
+ exp_manager(trainer, cfg['exp_manager'])
45
+
46
+ trainer.fit(model)
47
+
48
+ logging.info(f'Done!')
49
+
50
+ model.save_to(save_path="/ds/experiments/stt_en_quartznet15x5_v4.nemo")
51
+
52
+ logging.info(f'Done!')
53
+
54
+
55
+ if __name__ == '__main__':
56
+ main()
57
+
58
+ ```
quartznet15x5/2023-02-23_11-43-23/lightning_logs.txt ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Using 16bit native Automatic Mixed Precision (AMP)
2
+ GPU available: True (cuda), used: True
3
+ TPU available: False, using: 0 TPU cores
4
+ IPU available: False, using: 0 IPUs
5
+ HPU available: False, using: 0 HPUs
6
+ ----------------------------------------------------------------------------------------------------
7
+ distributed_backend=nccl
8
+ All distributed processes registered. Starting with 1 processes
9
+ ----------------------------------------------------------------------------------------------------
10
+
11
+ LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]
12
+
13
+ | Name | Type | Params
14
+ ------------------------------------------------------------------------
15
+ 0 | preprocessor | AudioToMelSpectrogramPreprocessor | 0
16
+ 1 | encoder | ConvASREncoder | 18.9 M
17
+ 2 | spec_augmentation | SpectrogramAugmentation | 0
18
+ 3 | _wer | WER | 0
19
+ 4 | decoder | ConvASRDecoder | 42.0 K
20
+ 5 | loss | CTCLoss | 0
21
+ ------------------------------------------------------------------------
22
+ 18.9 M Trainable params
23
+ 0 Non-trainable params
24
+ 18.9 M Total params
25
+ 37.873 Total estimated model params size (MB)
26
+ Epoch 0, global step 6790: 'val_loss' reached 17.07648 (best 17.07648), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=17.0765-epoch=0.ckpt' as top 3
27
+ Epoch 0, global step 13580: 'val_loss' reached 10.97846 (best 10.97846), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=10.9785-epoch=0.ckpt' as top 3
28
+ Epoch 0, global step 20370: 'val_loss' reached 9.71839 (best 9.71839), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=9.7184-epoch=0.ckpt' as top 3
29
+ Epoch 0, global step 27160: 'val_loss' reached 7.15682 (best 7.15682), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=7.1568-epoch=0.ckpt' as top 3
30
+ Epoch 1, global step 33951: 'val_loss' reached 6.46626 (best 6.46626), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=6.4663-epoch=1.ckpt' as top 3
31
+ Epoch 1, global step 40741: 'val_loss' reached 5.58261 (best 5.58261), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=5.5826-epoch=1.ckpt' as top 3
32
+ Epoch 1, global step 47531: 'val_loss' reached 5.77446 (best 5.58261), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=5.7745-epoch=1.ckpt' as top 3
33
+ Epoch 1, global step 54321: 'val_loss' reached 4.68531 (best 4.68531), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=4.6853-epoch=1.ckpt' as top 3
34
+ Epoch 2, global step 61112: 'val_loss' reached 4.32403 (best 4.32403), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=4.3240-epoch=2.ckpt' as top 3
35
+ Epoch 2, global step 67902: 'val_loss' reached 3.81663 (best 3.81663), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=3.8166-epoch=2.ckpt' as top 3
36
+ Epoch 2, global step 74692: 'val_loss' reached 4.16412 (best 3.81663), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=4.1641-epoch=2.ckpt' as top 3
37
+ Epoch 2, global step 81482: 'val_loss' reached 3.37959 (best 3.37959), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=3.3796-epoch=2.ckpt' as top 3
38
+ Epoch 3, global step 88273: 'val_loss' reached 3.45001 (best 3.37959), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=3.4500-epoch=3.ckpt' as top 3
39
+ Epoch 3, global step 95063: 'val_loss' reached 2.94031 (best 2.94031), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.9403-epoch=3.ckpt' as top 3
40
+ Epoch 3, global step 101853: 'val_loss' was not in top 3
41
+ Epoch 3, global step 108643: 'val_loss' reached 2.78752 (best 2.78752), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.7875-epoch=3.ckpt' as top 3
42
+ Epoch 4, global step 115434: 'val_loss' reached 2.69196 (best 2.69196), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.6920-epoch=4.ckpt' as top 3
43
+ Epoch 4, global step 122224: 'val_loss' reached 2.59559 (best 2.59559), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.5956-epoch=4.ckpt' as top 3
44
+ Epoch 4, global step 129014: 'val_loss' was not in top 3
45
+ Epoch 4, global step 135804: 'val_loss' reached 2.47542 (best 2.47542), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.4754-epoch=4.ckpt' as top 3
46
+ Epoch 5, global step 142595: 'val_loss' reached 2.34722 (best 2.34722), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.3472-epoch=5.ckpt' as top 3
47
+ Epoch 5, global step 149385: 'val_loss' reached 2.22968 (best 2.22968), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.2297-epoch=5.ckpt' as top 3
48
+ Epoch 5, global step 156175: 'val_loss' was not in top 3
49
+ Epoch 5, global step 162965: 'val_loss' reached 2.00784 (best 2.00784), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.0078-epoch=5.ckpt' as top 3
50
+ Epoch 6, global step 169756: 'val_loss' reached 2.13385 (best 2.00784), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.1338-epoch=6.ckpt' as top 3
51
+ Epoch 6, global step 176546: 'val_loss' reached 1.95768 (best 1.95768), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=1.9577-epoch=6.ckpt' as top 3
52
+ Epoch 6, global step 183336: 'val_loss' was not in top 3
53
+ Epoch 6, global step 190126: 'val_loss' reached 1.91314 (best 1.91314), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=1.9131-epoch=6.ckpt' as top 3
54
+ Epoch 7, global step 196917: 'val_loss' reached 1.90434 (best 1.90434), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=1.9043-epoch=7.ckpt' as top 3
55
+ Epoch 7, global step 203707: 'val_loss' reached 1.77183 (best 1.77183), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=1.7718-epoch=7.ckpt' as top 3
56
+ Epoch 7, global step 210497: 'val_loss' was not in top 3
57
+ Epoch 7, global step 217287: 'val_loss' reached 1.66200 (best 1.66200), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=1.6620-epoch=7.ckpt' as top 3
quartznet15x5/2023-02-23_11-43-23/stt_uk_ru_quartznet15x5.nemo ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adcfb10e9e02f00acac8ec952906693ac733a76bf30a5616fe4b63e4f7046f71
3
+ size 76400640
quartznet15x5/2023-02-23_11-43-23/stt_uk_ru_quartznet15x5.yaml ADDED
@@ -0,0 +1,287 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: &name "stt_uk_ru_quartznet15x5"
2
+
3
+ num_classes: &num_classes 40
4
+ sample_rate: &sample_rate 16000
5
+ repeat: &repeat 5
6
+ dropout: &dropout 0.0
7
+ separable: &separable true
8
+ labels: &labels [ " ", "а", "б", "в", "г", "ґ", "д", "е", "є", "ж", "з", "и", "і", "ї", "й", "к", "л", "м", "н",
9
+ "о", "п", "р", "с", "т", "у", "ф", "х", "ц", "ч", "ш", "щ", "ь", "ю", "я",
10
+ "ё", "ъ", "ы", "э", "'", "-" ]
11
+
12
+ model:
13
+ labels: *labels
14
+ sample_rate: *sample_rate
15
+ repeat: *repeat
16
+ dropout: *dropout
17
+ separable: *separable
18
+
19
+ train_ds:
20
+ manifest_filepath: /datasets/full_nemo_tarred/tarred_audio_manifest_utf8.json
21
+ sample_rate: *sample_rate
22
+ labels: *labels
23
+ batch_size: 64
24
+ trim_silence: True
25
+ max_duration: 9.7
26
+ min_duration: 1.0
27
+ shuffle: True
28
+ # shuffle_n: 2048
29
+ is_tarred: True
30
+ tarred_audio_filepaths: /datasets/full_nemo_tarred/audio__OP_0..299_CL_.tar
31
+ num_workers: 32
32
+ bucketing_strategy: "synced_randomized"
33
+ bucketing_batch_size: null
34
+
35
+ validation_ds:
36
+ manifest_filepath:
37
+ - /datasets/ds/cv_corpus_all_2022_12_31_uk/nemo_validation_manifest_docker.json
38
+ - /datasets/ds/cv_corpus_all_2022_12_31_ru/nemo_validation_manifest_docker.json
39
+ sample_rate: *sample_rate
40
+ labels: *labels
41
+ shuffle: False
42
+ batch_size: 32
43
+
44
+ preprocessor:
45
+ _target_: nemo.collections.asr.modules.AudioToMelSpectrogramPreprocessor
46
+ normalize: "per_feature"
47
+ window_size: 0.02
48
+ sample_rate: *sample_rate
49
+ window_stride: 0.01
50
+ window: "hann"
51
+ features: &n_mels 64
52
+ n_fft: 512
53
+ frame_splicing: 1
54
+ dither: 0.00001
55
+
56
+ spec_augment:
57
+ _target_: nemo.collections.asr.modules.SpectrogramAugmentation
58
+ rect_freq: 50
59
+ rect_masks: 5
60
+ rect_time: 120
61
+
62
+ encoder:
63
+ _target_: nemo.collections.asr.modules.ConvASREncoder
64
+ feat_in: *n_mels
65
+ activation: relu
66
+ conv_mask: true
67
+
68
+ jasper:
69
+ - dilation: [1]
70
+ dropout: *dropout
71
+ filters: 256
72
+ kernel: [33]
73
+ repeat: 1
74
+ residual: false
75
+ separable: *separable
76
+ stride: [2]
77
+
78
+ - dilation: [1]
79
+ dropout: *dropout
80
+ filters: 256
81
+ kernel: [33]
82
+ repeat: *repeat
83
+ residual: true
84
+ separable: *separable
85
+ stride: [1]
86
+
87
+ - dilation: [1]
88
+ dropout: *dropout
89
+ filters: 256
90
+ kernel: [33]
91
+ repeat: *repeat
92
+ residual: true
93
+ separable: *separable
94
+ stride: [1]
95
+
96
+ - dilation: [1]
97
+ dropout: *dropout
98
+ filters: 256
99
+ kernel: [33]
100
+ repeat: *repeat
101
+ residual: true
102
+ separable: *separable
103
+ stride: [1]
104
+
105
+ - dilation: [1]
106
+ dropout: *dropout
107
+ filters: 256
108
+ kernel: [39]
109
+ repeat: *repeat
110
+ residual: true
111
+ separable: *separable
112
+ stride: [1]
113
+
114
+ - dilation: [1]
115
+ dropout: *dropout
116
+ filters: 256
117
+ kernel: [39]
118
+ repeat: *repeat
119
+ residual: true
120
+ separable: *separable
121
+ stride: [1]
122
+
123
+ - dilation: [1]
124
+ dropout: *dropout
125
+ filters: 256
126
+ kernel: [39]
127
+ repeat: *repeat
128
+ residual: true
129
+ separable: *separable
130
+ stride: [1]
131
+
132
+ - dilation: [1]
133
+ dropout: *dropout
134
+ filters: 512
135
+ kernel: [51]
136
+ repeat: *repeat
137
+ residual: true
138
+ separable: *separable
139
+ stride: [1]
140
+
141
+ - dilation: [1]
142
+ dropout: *dropout
143
+ filters: 512
144
+ kernel: [51]
145
+ repeat: *repeat
146
+ residual: true
147
+ separable: *separable
148
+ stride: [1]
149
+
150
+ - dilation: [1]
151
+ dropout: *dropout
152
+ filters: 512
153
+ kernel: [51]
154
+ repeat: *repeat
155
+ residual: true
156
+ separable: *separable
157
+ stride: [1]
158
+
159
+ - dilation: [1]
160
+ dropout: *dropout
161
+ filters: 512
162
+ kernel: [63]
163
+ repeat: *repeat
164
+ residual: true
165
+ separable: *separable
166
+ stride: [1]
167
+
168
+ - dilation: [1]
169
+ dropout: *dropout
170
+ filters: 512
171
+ kernel: [63]
172
+ repeat: *repeat
173
+ residual: true
174
+ separable: *separable
175
+ stride: [1]
176
+
177
+ - dilation: [1]
178
+ dropout: *dropout
179
+ filters: 512
180
+ kernel: [63]
181
+ repeat: *repeat
182
+ residual: true
183
+ separable: *separable
184
+ stride: [1]
185
+
186
+ - dilation: [1]
187
+ dropout: *dropout
188
+ filters: 512
189
+ kernel: [75]
190
+ repeat: *repeat
191
+ residual: true
192
+ separable: *separable
193
+ stride: [1]
194
+
195
+ - dilation: [1]
196
+ dropout: *dropout
197
+ filters: 512
198
+ kernel: [75]
199
+ repeat: *repeat
200
+ residual: true
201
+ separable: *separable
202
+ stride: [1]
203
+
204
+ - dilation: [1]
205
+ dropout: *dropout
206
+ filters: 512
207
+ kernel: [75]
208
+ repeat: *repeat
209
+ residual: true
210
+ separable: *separable
211
+ stride: [1]
212
+
213
+ - dilation: [2]
214
+ dropout: *dropout
215
+ filters: 512
216
+ kernel: [87]
217
+ repeat: 1
218
+ residual: false
219
+ separable: *separable
220
+ stride: [1]
221
+
222
+ - dilation: [1]
223
+ dropout: *dropout
224
+ filters: &enc_filters 1024
225
+ kernel: [1]
226
+ repeat: 1
227
+ residual: false
228
+ stride: [1]
229
+
230
+ decoder:
231
+ _target_: nemo.collections.asr.modules.ConvASRDecoder
232
+ feat_in: *enc_filters
233
+ num_classes: *num_classes
234
+ vocabulary: *labels
235
+
236
+ optim:
237
+ name: novograd
238
+ lr: 0.001
239
+ betas: [0.8, 0.5]
240
+ weight_decay: 0.0001
241
+
242
+ sched:
243
+ name: CosineAnnealing
244
+ warmup_steps: null
245
+ warmup_ratio: null
246
+ min_lr: 0.0
247
+ last_epoch: -1
248
+
249
+ # pytorch lightning args
250
+ monitor: val_loss
251
+ reduce_on_plateau: false
252
+
253
+ trainer:
254
+ # gpus: -1 # number of gpus was configured so validation will run at the end of the training epoch..
255
+ accelerator: gpu
256
+ devices: -1 # number of gpus
257
+ max_epochs: 50
258
+ max_steps: -1 # computed at runtime if not set
259
+ num_nodes: 1
260
+ strategy: ddp
261
+ accumulate_grad_batches: 1
262
+ enable_checkpointing: False # Provided by exp_manager
263
+ logger: False # Provided by exp_manager
264
+ log_every_n_steps: 1 # Interval of logging.
265
+ val_check_interval: 0.25 # Set to 0.25 to check 4 times per epoch, or an int for number of iterations
266
+ benchmark: false # має бути false для моделей із мовним введенням змінної довжини, оскільки це сповільнює навчання
267
+ precision: 16 # Should be set to 16 for O1 and O2 to enable the AMP.
268
+
269
+ exp_manager:
270
+ exp_dir: "/ds/experiments"
271
+ name: *name
272
+ create_tensorboard_logger: False
273
+ create_checkpoint_callback: True
274
+ # checkpoint_callback_params:
275
+ # monitor: "val_wer"
276
+ # mode: "min"
277
+ create_wandb_logger: False
278
+ wandb_logger_kwargs:
279
+ name: null
280
+ project: null
281
+
282
+ hydra:
283
+ run:
284
+ dir: .
285
+ job_logging:
286
+ root:
287
+ handlers: null
quartznet15x5/2023-02-23_11-43-23/terminal_output.txt ADDED
@@ -0,0 +1,263 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ HYDRA_FULL_ERROR=1 python /ds/training_stt_en_quartznet15x5_v4.py
2
+ [NeMo W 2023-02-23 11:38:01 experimental:27] Module <class 'nemo.collections.asr.models.audio_to_audio_model.AudioToAudioModel'> is experimental, not ready for production and is not fully supported. Use at your own risk.
3
+ [NeMo W 2023-02-23 11:38:02 experimental:27] Module <class 'nemo.collections.asr.modules.audio_modules.SpectrogramToMultichannelFeatures'> is experimental, not ready for production and is not fully supported. Use at your own risk.
4
+ [NeMo W 2023-02-23 11:38:02 nemo_logging:349] /usr/local/lib/python3.8/dist-packages/torch/jit/annotations.py:309: UserWarning: TorchScript will treat type annotations of Tensor dtype-specific subtypes as if they are normal Tensors. dtype constraints are not enforced in compilation either.
5
+ warnings.warn("TorchScript will treat type annotations of Tensor "
6
+
7
+ [NeMo W 2023-02-23 11:38:02 experimental:27] Module <class 'nemo.collections.asr.data.audio_to_audio.BaseAudioDataset'> is experimental, not ready for production and is not fully supported. Use at your own risk.
8
+ [NeMo W 2023-02-23 11:38:02 experimental:27] Module <class 'nemo.collections.asr.data.audio_to_audio.AudioToTargetDataset'> is experimental, not ready for production and is not fully supported. Use at your own risk.
9
+ [NeMo W 2023-02-23 11:38:02 experimental:27] Module <class 'nemo.collections.asr.data.audio_to_audio.AudioToTargetWithReferenceDataset'> is experimental, not ready for production and is not fully supported. Use at your own risk.
10
+ [NeMo W 2023-02-23 11:38:02 experimental:27] Module <class 'nemo.collections.asr.data.audio_to_audio.AudioToTargetWithEmbeddingDataset'> is experimental, not ready for production and is not fully supported. Use at your own risk.
11
+ [NeMo W 2023-02-23 11:38:02 experimental:27] Module <class 'nemo.collections.asr.models.enhancement_models.EncMaskDecAudioToAudioModel'> is experimental, not ready for production and is not fully supported. Use at your own risk.
12
+ [NeMo W 2023-02-23 11:38:02 nemo_logging:349] /usr/local/lib/python3.8/dist-packages/hydra/_internal/hydra.py:119: UserWarning: Future Hydra versions will no longer change working directory at job runtime by default.
13
+ See https://hydra.cc/docs/next/upgrades/1.1_to_1.2/changes_to_job_working_dir/ for more information.
14
+ ret = run_job(
15
+
16
+ [NeMo I 2023-02-23 11:38:02 training_stt_en_quartznet15x5_v4:21] Конфігурація Hydra
17
+ [NeMo I 2023-02-23 11:38:02 cloud:56] Found existing object /root/.cache/torch/NeMo/NeMo_1.15.0/stt_en_quartznet15x5/16661021d16e679bdfd97a2a03944c49/stt_en_quartznet15x5.nemo.
18
+ [NeMo I 2023-02-23 11:38:02 cloud:62] Re-using file from: /root/.cache/torch/NeMo/NeMo_1.15.0/stt_en_quartznet15x5/16661021d16e679bdfd97a2a03944c49/stt_en_quartznet15x5.nemo
19
+ [NeMo I 2023-02-23 11:38:02 common:913] Instantiating model from pre-trained checkpoint
20
+ [NeMo W 2023-02-23 11:38:03 modelPT:156] If you intend to do training or fine-tuning, please call the ModelPT.setup_training_data() method and provide a valid configuration file to setup the train data loader.
21
+ Train config :
22
+ manifest_filepath: /data2/voices/train_1k.json
23
+ sample_rate: 16000
24
+ labels:
25
+ - ' '
26
+ - a
27
+ - b
28
+ - c
29
+ - d
30
+ - e
31
+ - f
32
+ - g
33
+ - h
34
+ - i
35
+ - j
36
+ - k
37
+ - l
38
+ - m
39
+ - 'n'
40
+ - o
41
+ - p
42
+ - q
43
+ - r
44
+ - s
45
+ - t
46
+ - u
47
+ - v
48
+ - w
49
+ - x
50
+ - 'y'
51
+ - z
52
+ - ''''
53
+ batch_size: 32
54
+ trim_silence: true
55
+ max_duration: 16.7
56
+ shuffle: true
57
+ is_tarred: false
58
+ tarred_audio_filepaths: /asr_set_1.2/train/train_{0..1023}.tar
59
+ num_workers: 20
60
+
61
+ [NeMo W 2023-02-23 11:38:03 modelPT:163] If you intend to do validation, please call the ModelPT.setup_validation_data() or ModelPT.setup_multiple_validation_data() method and provide a valid configuration file to setup the validation data loader(s).
62
+ Validation config :
63
+ manifest_filepath: /data2/voices/train_1k_samp.json
64
+ sample_rate: 16000
65
+ labels:
66
+ - ' '
67
+ - a
68
+ - b
69
+ - c
70
+ - d
71
+ - e
72
+ - f
73
+ - g
74
+ - h
75
+ - i
76
+ - j
77
+ - k
78
+ - l
79
+ - m
80
+ - 'n'
81
+ - o
82
+ - p
83
+ - q
84
+ - r
85
+ - s
86
+ - t
87
+ - u
88
+ - v
89
+ - w
90
+ - x
91
+ - 'y'
92
+ - z
93
+ - ''''
94
+ batch_size: 32
95
+ shuffle: false
96
+
97
+ [NeMo I 2023-02-23 11:38:03 features:267] PADDING: 16
98
+ [NeMo I 2023-02-23 11:38:04 save_restore_connector:243] Model EncDecCTCModel was successfully restored from /root/.cache/torch/NeMo/NeMo_1.15.0/stt_en_quartznet15x5/16661021d16e679bdfd97a2a03944c49/stt_en_quartznet15x5.nemo.
99
+ [NeMo I 2023-02-23 11:38:04 ctc_models:294] Changed decoder to output to [' ', 'а', 'б', 'в', 'г', 'ґ', 'д', 'е', 'є', 'ж', 'з', 'и', 'і', 'ї', 'й', 'к', 'л', 'м', 'н', 'о', 'п', 'р', 'с', 'т', 'у', 'ф', 'х', 'ц', 'ч', 'ш', 'щ', 'ь', 'ю', 'я', 'ё', 'ъ', 'ы', 'э', "'", '-'] vocabulary.
100
+ [NeMo W 2023-02-23 11:38:04 modelPT:505] Trainer wasn't specified in model constructor. Make sure that you really wanted it.
101
+ [NeMo I 2023-02-23 11:38:04 modelPT:616] Optimizer config = Novograd (
102
+ Parameter Group 0
103
+ amsgrad: False
104
+ betas: [0.8, 0.5]
105
+ eps: 1e-08
106
+ grad_averaging: False
107
+ lr: 0.001
108
+ weight_decay: 0.0001
109
+ )
110
+ [NeMo W 2023-02-23 11:38:04 lr_scheduler:890] Neither `max_steps` nor `iters_per_batch` were provided to `optim.sched`, cannot compute effective `max_steps` !
111
+ Scheduler will not be instantiated !
112
+ [NeMo W 2023-02-23 11:38:04 audio_to_text_dataset:56] `labels` is explicitly provided to the data loader, and is different from the `labels` provided at the model level config.
113
+ If this is incorrect, please set the dataloader's `labels` to None.
114
+ [NeMo I 2023-02-23 11:43:15 collections:193] Dataset loaded with 1738252 files totalling 1645.74 hours
115
+ [NeMo I 2023-02-23 11:43:15 collections:194] 18548 files were filtered totalling 61.11 hours
116
+ [NeMo W 2023-02-23 11:43:15 ctc_models:461] Model Trainer was not set before constructing the dataset, incorrect number of training batches will be used. Please set the trainer and rebuild the dataset.
117
+ [NeMo W 2023-02-23 11:43:15 audio_to_text_dataset:56] `labels` is explicitly provided to the data loader, and is different from the `labels` provided at the model level config.
118
+ If this is incorrect, please set the dataloader's `labels` to None.
119
+ [NeMo I 2023-02-23 11:43:23 collections:193] Dataset loaded with 11373 files totalling 12.63 hours
120
+ [NeMo I 2023-02-23 11:43:23 collections:194] 0 files were filtered totalling 0.00 hours
121
+ Using 16bit native Automatic Mixed Precision (AMP)
122
+ GPU available: True (cuda), used: True
123
+ TPU available: False, using: 0 TPU cores
124
+ IPU available: False, using: 0 IPUs
125
+ HPU available: False, using: 0 HPUs
126
+ [NeMo I 2023-02-23 11:43:23 exp_manager:361] Experiments will be logged at /ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23
127
+ Initializing distributed: GLOBAL_RANK: 0, MEMBER: 1/1
128
+ ----------------------------------------------------------------------------------------------------
129
+ distributed_backend=nccl
130
+ All distributed processes registered. Starting with 1 processes
131
+ ----------------------------------------------------------------------------------------------------
132
+
133
+ LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]
134
+ [NeMo I 2023-02-23 11:43:24 modelPT:616] Optimizer config = Novograd (
135
+ Parameter Group 0
136
+ amsgrad: False
137
+ betas: [0.8, 0.5]
138
+ eps: 1e-08
139
+ grad_averaging: False
140
+ lr: 0.001
141
+ weight_decay: 0.0001
142
+ )
143
+ [NeMo I 2023-02-23 11:43:24 lr_scheduler:910] Scheduler "<nemo.core.optim.lr_scheduler.CosineAnnealing object at 0x7f6e52320a60>"
144
+ will be used during training (effective maximum steps = 1358050) -
145
+ Parameters :
146
+ (warmup_steps: null
147
+ warmup_ratio: null
148
+ min_lr: 0.0
149
+ last_epoch: -1
150
+ max_steps: 1358050
151
+ )
152
+
153
+ | Name | Type | Params
154
+ ------------------------------------------------------------------------
155
+ 0 | preprocessor | AudioToMelSpectrogramPreprocessor | 0
156
+ 1 | encoder | ConvASREncoder | 18.9 M
157
+ 2 | spec_augmentation | SpectrogramAugmentation | 0
158
+ 3 | _wer | WER | 0
159
+ 4 | decoder | ConvASRDecoder | 42.0 K
160
+ 5 | loss | CTCLoss | 0
161
+ ------------------------------------------------------------------------
162
+ 18.9 M Trainable params
163
+ 0 Non-trainable params
164
+ 18.9 M Total params
165
+ 37.873 Total estimated model params size (MB)
166
+ Sanity Checking: 0it [00:00, ?it/s][NeMo W 2023-02-23 11:43:24 nemo_logging:349] /usr/local/lib/python3.8/dist-packages/pytorch_lightning/trainer/connectors/data_connector.py:224: PossibleUserWarning: The dataloader, val_dataloader 0, does not have many workers which may be a bottleneck. Consider increasing the value of the `num_workers` argument` (try 32 which is the number of cpus on this machine) in the `DataLoader` init to improve performance.
167
+ rank_zero_warn(
168
+
169
+ Sanity Checking DataLoader 0: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 2/2 [00:00<00:00, 2.72it/s][NeMo W 2023-02-23 11:43:25 nemo_logging:349] /usr/local/lib/python3.8/dist-packages/pytorch_lightning/trainer/connectors/logger_connector/result.py:537: PossibleUserWarning: It is recommended to use `self.log('global_step', ..., sync_dist=True)` when logging on epoch level in distributed setting to accumulate the metric across devices.
170
+ warning_cache.warn(
171
+
172
+ [NeMo W 2023-02-23 11:43:25 nemo_logging:349] /usr/local/lib/python3.8/dist-packages/pytorch_lightning/trainer/connectors/logger_connector/result.py:537: PossibleUserWarning: It is recommended to use `self.log('val_loss', ..., sync_dist=True)` when logging on epoch level in distributed setting to accumulate the metric across devices.
173
+ warning_cache.warn(
174
+
175
+ [NeMo W 2023-02-23 11:43:25 nemo_logging:349] /usr/local/lib/python3.8/dist-packages/pytorch_lightning/trainer/connectors/logger_connector/result.py:537: PossibleUserWarning: It is recommended to use `self.log('val_wer', ..., sync_dist=True)` when logging on epoch level in distributed setting to accumulate the metric across devices.
176
+ warning_cache.warn(
177
+
178
+ [NeMo W 2023-02-23 11:43:25 nemo_logging:349] /usr/local/lib/python3.8/dist-packages/lightning_lite/utilities/data.py:63: UserWarning: Your `IterableDataset` has `__len__` defined. In combination with multi-process data loading (when num_workers > 1), `__len__` could be inaccurate if each worker is not configured independently to avoid having duplicate data.
179
+ rank_zero_warn(
180
+
181
+ Epoch 0: 0%| | 0/28585 [00:00<?, ?it/s][W reducer.cpp:1298] Warning: find_unused_parameters=True was specified in DDP constructor, but did not find any unused parameters in the forward pass. This flag results in an extra traversal of the autograd graph every iteration, which can adversely affect performance. If your model indeed never has any unused parameters in the forward pass, consider turning this flag off. Note that this warning may be a false positive if your model has flow control causing later iterations to have unused parameters. (function operator())
182
+ [NeMo W 2023-02-23 11:43:28 nemo_logging:349] /usr/local/lib/python3.8/dist-packages/torch/optim/lr_scheduler.py:138: UserWarning: Detected call of `lr_scheduler.step()` before `optimizer.step()`. In PyTorch 1.1.0 and later, you should call them in the opposite order: `optimizer.step()` before `lr_scheduler.step()`. Failure to do this will result in PyTorch skipping the first value of the learning rate schedule. See more details at https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate
183
+ warnings.warn("Detected call of `lr_scheduler.step()` before `optimizer.step()`. "
184
+
185
+ Epoch 0: 25%|████████████████████████████████████▍ | 7146/28585 [34:16<1:42:51, 3.47it/s, loss=26.1
186
+ Epoch 0, global step 6790: 'val_loss' reached 17.07648 (best 17.07648), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=17.0765-epoch=0.ckpt' as top 3
187
+ Epoch 0: 50%|███████████████████████████████████████████████████████████████████████▍ | 14292/28585 [1:08:40<1:08:40, 3.47it/s, loss=26.7
188
+ Epoch 0, global step 13580: 'val_loss' reached 10.97846 (best 10.97846), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=10.9785-epoch=0.ckpt' as top 3
189
+ Epoch 0: 55%|███████████████████████████████████████████████████████████████████████████████▋ | 15719/28585 [1:16:06<1:02:17, 3.44it/s, loss=20]
190
+ Epoch 0: 64%|█████████████████████████████████████████████████████████████████████████████████████████████▎ | 18394/28585 [1:28:51<49:13, 3.45it/s, loss=13.2]
191
+ Epoch 0: 75%|████████████████████████████████████████████████████████████████████████████████████████████████████████████▋ | 21438/28585 [1:43:49<34:36, 3.44it/s, loss=11.4
192
+ Epoch 0, global step 20370: 'val_loss' reached 9.71839 (best 9.71839), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=9.7184-epoch=0.ckpt' as top 3
193
+ Epoch 0: 75%|████████████████████████████████████████████████████████████████████████████████████████████████████████████▉ | 21486/28585 [1:44:07<34:24, 3.44it/s, loss=10.2]
194
+ Epoch 0: 91%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▎ | 26080/28585 [2:05:48<12:05, 3.45it/s, loss=21.6]
195
+ Epoch 0: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▉| 28584/28585 [2:18:41<00:00, 3.44it/s, loss=28.5
196
+ Epoch 0, global step 27160: 'val_loss' reached 7.15682 (best 7.15682), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=7.1568-epoch=0.ckpt' as top 3
197
+ Epoch 1: 25%|████████████████████████████████████▍ | 7146/28585 [34:01<1:42:03, 3.50it/s, loss=17.7
198
+ Epoch 1, global step 33951: 'val_loss' reached 6.46626 (best 6.46626), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=6.4663-epoch=1.ckpt' as top 3
199
+ Epoch 1: 50%|███████████████████████████████████████████████████████████████████████▍ | 14292/28585 [1:08:25<1:08:25, 3.48it/s, loss=20.3
200
+ Epoch 1, global step 40741: 'val_loss' reached 5.58261 (best 5.58261), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=5.5826-epoch=1.ckpt' as top 3
201
+ Epoch 1: 75%|████████████████████████████████████████████████████████████████████████████████████████████████████████████▋ | 21438/28585 [1:43:30<34:30, 3.45it/s, loss=8.82
202
+ Epoch 1, global step 47531: 'val_loss' reached 5.77446 (best 5.58261), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=5.7745-epoch=1.ckpt' as top 3
203
+ Epoch 1: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▉| 28584/28585 [2:18:16<00:00, 3.45it/s, loss=23.5
204
+ Epoch 1, global step 54321: 'val_loss' reached 4.68531 (best 4.68531), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=4.6853-epoch=1.ckpt' as top 3
205
+ Epoch 2: 25%|████████████████████████████████████▍ | 7146/28585 [34:06<1:42:18, 3.49it/s, loss=15.2
206
+ Epoch 2, global step 61112: 'val_loss' reached 4.32403 (best 4.32403), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=4.3240-epoch=2.ckpt' as top 3
207
+ Epoch 2: 50%|███████████████████████████████████████████████████████████████████████▍ | 14292/28585 [1:08:31<1:08:32, 3.48it/s, loss=16.2
208
+ Epoch 2, global step 67902: 'val_loss' reached 3.81663 (best 3.81663), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=3.8166-epoch=2.ckpt' as top 3
209
+ Epoch 2: 75%|███████████████████████████████████████████████████████████████████████████���████████████████████████████████▋ | 21438/28585 [1:43:50<34:37, 3.44it/s, loss=6.72
210
+ Epoch 2, global step 74692: 'val_loss' reached 4.16412 (best 3.81663), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=4.1641-epoch=2.ckpt' as top 3
211
+ Epoch 2: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▉| 28584/28585 [2:18:39<00:00, 3.44it/s, loss=21.3
212
+ Epoch 2, global step 81482: 'val_loss' reached 3.37959 (best 3.37959), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=3.3796-epoch=2.ckpt' as top 3
213
+ Epoch 3: 25%|████████████████████████████████████▍ | 7146/28585 [33:58<1:41:54, 3.51it/s, loss=13.6
214
+ Epoch 3, global step 88273: 'val_loss' reached 3.45001 (best 3.37959), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=3.4500-epoch=3.ckpt' as top 3
215
+ Epoch 3: 50%|███████████████████████████████████████████████████████████████████████▍ | 14292/28585 [1:08:41<1:08:41, 3.47it/s, loss=14.3
216
+ Epoch 3, global step 95063: 'val_loss' reached 2.94031 (best 2.94031), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.9403-epoch=3.ckpt' as top 3
217
+ Epoch 3: 75%|████████████████████████████████████████████████████████████████████████████████████████████████████████████▋ | 21438/28585 [1:43:51<34:37, 3.44it/s, loss=6.29
218
+ Epoch 3, global step 101853: 'val_loss' was not in top 3
219
+ Epoch 3: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▉| 28584/28585 [2:18:30<00:00, 3.44it/s, loss=18.6
220
+ Epoch 3, global step 108643: 'val_loss' reached 2.78752 (best 2.78752), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.7875-epoch=3.ckpt' as top 3
221
+ Epoch 4: 25%|████████████████████████████████████▍ | 7146/28585 [34:20<1:43:01, 3.47it/s, loss=12.4
222
+ Epoch 4, global step 115434: 'val_loss' reached 2.69196 (best 2.69196), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.6920-epoch=4.ckpt' as top 3
223
+ Epoch 4: 50%|███████████████████████████████████████████████████████████████████████▍ | 14292/28585 [1:08:58<1:08:58, 3.45it/s, loss=13.1
224
+ Epoch 4, global step 122224: 'val_loss' reached 2.59559 (best 2.59559), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.5956-epoch=4.ckpt' as top 3
225
+ Epoch 4: 75%|████████████████████████████████████████████████████████████████████████████████████████████████████████████▋ | 21438/28585 [1:44:16<34:45, 3.43it/s, loss=4.96
226
+ Epoch 4, global step 129014: 'val_loss' was not in top 3
227
+ Epoch 4: 100%|███████████████████████████��████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▉| 28584/28585 [2:18:56<00:00, 3.43it/s, loss=16.7
228
+ Epoch 4, global step 135804: 'val_loss' reached 2.47542 (best 2.47542), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.4754-epoch=4.ckpt' as top 3
229
+ Epoch 5: 25%|████████████████████████████████████▍ | 7146/28585 [34:03<1:42:10, 3.50it/s, loss=11.3
230
+ Epoch 5, global step 142595: 'val_loss' reached 2.34722 (best 2.34722), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.3472-epoch=5.ckpt' as top 3
231
+ Epoch 5: 50%|███████████████████████████████████████████████████████████████████████▍ | 14292/28585 [1:08:38<1:08:38, 3.47it/s, loss=12.5
232
+ Epoch 5, global step 149385: 'val_loss' reached 2.22968 (best 2.22968), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.2297-epoch=5.ckpt' as top 3
233
+ Epoch 5: 75%|████████████████████████████████████████████████████████████████████████████████████████████████████████████▋ | 21438/28585 [1:43:57<34:39, 3.44it/s, loss=5.45
234
+ Epoch 5, global step 156175: 'val_loss' was not in top 3
235
+ Epoch 5: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▉| 28584/28585 [2:18:35<00:00, 3.44it/s, loss=17.1
236
+ Epoch 5, global step 162965: 'val_loss' reached 2.00784 (best 2.00784), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.0078-epoch=5.ckpt' as top 3
237
+ Epoch 6: 25%|████████████████████████████████████▍ | 7146/28585 [34:29<1:43:30, 3.45it/s, loss=10.7
238
+ Epoch 6, global step 169756: 'val_loss' reached 2.13385 (best 2.00784), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=2.1338-epoch=6.ckpt' as top 3
239
+ Epoch 6: 50%|███████████████████████████████████████████████████████████████████████▍ | 14292/28585 [1:08:50<1:08:50, 3.46it/s, loss=11.3
240
+ Epoch 6, global step 176546: 'val_loss' reached 1.95768 (best 1.95768), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=1.9577-epoch=6.ckpt' as top 3
241
+ Epoch 6: 75%|████████████████████████████████████████████████████████████████████████████████████████████████████████████▋ | 21438/28585 [1:44:03<34:41, 3.43it/s, loss=4.76
242
+ Epoch 6, global step 183336: 'val_loss' was not in top 3
243
+ Epoch 6: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▉| 28584/28585 [2:18:34<00:00, 3.44it/s, loss=16.4
244
+ Epoch 6, global step 190126: 'val_loss' reached 1.91314 (best 1.91314), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=1.9131-epoch=6.ckpt' as top 3
245
+ Epoch 7: 25%|████████████████████████████████████▍ | 7146/28585 [34:35<1:43:47, 3.44it/s, loss=10.9
246
+ Epoch 7, global step 196917: 'val_loss' reached 1.90434 (best 1.90434), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=1.9043-epoch=7.ckpt' as top 3
247
+ Epoch 7: 50%|███████████████████████████████████████████████████████████████████████▍ | 14292/28585 [1:09:00<1:09:00, 3.45it/s, loss=11.6
248
+ Epoch 7, global step 203707: 'val_loss' reached 1.77183 (best 1.77183), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=1.7718-epoch=7.ckpt' as top 3
249
+ Epoch 7: 58%|███████████████████████████████████████████████████████████████████████████████████▉ | 16559/28585 [1:20:27<58:25, 3.43it/s, loss=10.9]
250
+ Epoch 7: 69%|████████████████████████████████████████████████████████████████████████████████████████████████████▋ | 19848/28585 [1:36:38<42:32, 3.42it/s, loss=11.2]
251
+ Epoch 7: 75%|████████████████████████████████████████████████████████████████████████████████████████████████████████████▋ | 21438/28585 [1:45:22<35:07, 3.39it/s, loss=4.96
252
+ Epoch 7, global step 210497: 'val_loss' was not in top 3
253
+ Epoch 7: 98%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████ | 28014/28585 [2:17:05<02:47, 3.41it/s, loss=12.1]
254
+ Epoch 7: 99%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▋ | 28320/28585 [2:18:51<01:17, 3.40it/s, loss=14.6]
255
+ Epoch 7: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▉| 28584/28585 [2:20:34<00:00, 3.39it/s, loss=14.6
256
+ Epoch 7, global step 217287: 'val_loss' reached 1.66200 (best 1.66200), saving model to '/ds/experiments/stt_en_quartznet15x5_v4/2023-02-23_11-43-23/checkpoints/stt_en_quartznet15x5_v4--val_loss=1.6620-epoch=7.ckpt' as top 3
257
+ Epoch 8: 0%|▋ | 140/28585 [00:37<2:06:59, 3.73it/s, loss=10.4]
258
+ ^C[NeMo W 2023-02-24 06:15:00 nemo_logging:349] /usr/local/lib/python3.8/dist-packages/pytorch_lightning/trainer/call.py:48: UserWarning: Detected KeyboardInterrupt, attempting graceful shutdown...
259
+ rank_zero_warn("Detected KeyboardInterrupt, attempting graceful shutdown...")
260
+
261
+ [NeMo I 2023-02-24 06:15:00 training_stt_en_quartznet15x5_v4:45] Done!
262
+ [NeMo I 2023-02-24 06:15:00 training_stt_en_quartznet15x5_v4:49] Done!
263
+ root@bd72308fa83d:/workspace/nemo#