Instructions to use u107/nemo with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- NeMo
How to use u107/nemo with NeMo:
# tag did not correspond to a valid NeMo domain.
- Notebooks
- Google Colab
- Kaggle
Vadim commited on
Commit ·
e484d37
1
Parent(s): 9fba329
add QuartzNet15x5
Browse files- README.md +10 -0
- quartznet15x5/2023-02-23_11-43-23/QuartzNet15x5--val_loss=1.6620-epoch=7-last.ckpt +3 -0
- quartznet15x5/2023-02-23_11-43-23/README.md +58 -0
- quartznet15x5/2023-02-23_11-43-23/lightning_logs.txt +57 -0
- quartznet15x5/2023-02-23_11-43-23/stt_uk_ru_quartznet15x5.nemo +3 -0
- quartznet15x5/2023-02-23_11-43-23/stt_uk_ru_quartznet15x5.yaml +287 -0
- quartznet15x5/2023-02-23_11-43-23/terminal_output.txt +263 -0
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#
|