Vadim commited on
Commit
8e1f50a
·
1 Parent(s): 75db244

update README.md

Browse files
Files changed (1) hide show
  1. README.md +36 -0
README.md CHANGED
@@ -58,3 +58,39 @@ tags:
58
  - _last checkpoint:_ ```jasper10x5/2023-02-22_06-46-21/Jasper10x5--val_loss=1.7606-epoch=4-last.ckpt```
59
  - _Dataset train:_ **1738252** files totalling **1645.74** hours
60
  - _Dataset validation:_ **11373** files totalling **12.63** hours
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
58
  - _last checkpoint:_ ```jasper10x5/2023-02-22_06-46-21/Jasper10x5--val_loss=1.7606-epoch=4-last.ckpt```
59
  - _Dataset train:_ **1738252** files totalling **1645.74** hours
60
  - _Dataset validation:_ **11373** files totalling **12.63** hours
61
+
62
+
63
+ Використання:
64
+ ```python
65
+ from huggingface_hub import hf_hub_download
66
+ from nemo.collections.asr.models import ASRModel
67
+ from omegaconf import OmegaConf
68
+
69
+
70
+ # filename_model = 'quartznet15x5/2023-02-23_11-43-23/stt_uk_ru_quartznet15x5.nemo'
71
+ # filename_model = 'quartznet15x5/2023-04-04_02-57-02/stt_uk_quartznet15x5.nemo'
72
+ # filename_model = 'jasper10x5/2023-02-22_06-46-21/stt_uk_ru_jasper10x5.nemo'
73
+ filename_model = 'jasper10x5/2023-04-07_05-01-29/stt_uk_jasper10x5.nemo'
74
+
75
+ model_path = hf_hub_download(
76
+ repo_id="u107/nemo",
77
+ filename=filename_model,
78
+ repo_type="model",
79
+ cache_dir='/content',
80
+ use_auth_token='hf_...')
81
+
82
+ nemo_model = ASRModel.restore_from(
83
+ restore_path=model_path,
84
+ # map_location='CPU',
85
+ # return_config=True,
86
+ ) # type: ASRModel
87
+
88
+ # зберегти файл налаштувань
89
+ OmegaConf.save(nemo_model.cfg, 'model_config.yaml')
90
+
91
+ # відобразити версію nemo на якій було навчання
92
+ print(nemo_model.cfg.nemo_version)
93
+
94
+ # відобразити алфавіт
95
+ print(nemo_model.cfg.labels)
96
+ ```