GetmanY1 commited on
Commit
013e89c
·
verified ·
1 Parent(s): e8330cd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +76 -164
README.md CHANGED
@@ -1,199 +1,111 @@
1
  ---
 
2
  library_name: transformers
3
- tags: []
 
 
 
 
 
 
4
  ---
 
5
 
6
- # Model Card for Model ID
7
 
8
- <!-- Provide a quick summary of what the model is/does. -->
9
 
 
10
 
 
11
 
12
- ## Model Details
13
 
14
- ### Model Description
15
 
16
- <!-- Provide a longer summary of what this model is. -->
17
 
18
- This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
19
 
20
- - **Developed by:** [More Information Needed]
21
- - **Funded by [optional]:** [More Information Needed]
22
- - **Shared by [optional]:** [More Information Needed]
23
- - **Model type:** [More Information Needed]
24
- - **Language(s) (NLP):** [More Information Needed]
25
- - **License:** [More Information Needed]
26
- - **Finetuned from model [optional]:** [More Information Needed]
27
 
28
- ### Model Sources [optional]
29
 
30
- <!-- Provide the basic links for the model. -->
31
 
32
- - **Repository:** [More Information Needed]
33
- - **Paper [optional]:** [More Information Needed]
34
- - **Demo [optional]:** [More Information Needed]
35
 
36
- ## Uses
37
 
38
- <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
 
40
- ### Direct Use
41
 
42
- <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
 
 
43
 
44
- [More Information Needed]
45
 
46
- ### Downstream Use [optional]
47
 
48
- <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
 
50
- [More Information Needed]
51
 
52
- ### Out-of-Scope Use
53
 
54
- <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
 
56
- [More Information Needed]
 
 
 
 
 
 
 
 
 
 
 
 
57
 
58
- ## Bias, Risks, and Limitations
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
59
 
60
- <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
 
62
- [More Information Needed]
 
 
 
 
 
 
63
 
64
- ### Recommendations
65
 
66
- <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
67
 
68
- Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
 
69
 
70
- ## How to Get Started with the Model
71
 
72
- Use the code below to get started with the model.
 
73
 
74
- [More Information Needed]
75
-
76
- ## Training Details
77
-
78
- ### Training Data
79
-
80
- <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
-
82
- [More Information Needed]
83
-
84
- ### Training Procedure
85
-
86
- <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
-
88
- #### Preprocessing [optional]
89
-
90
- [More Information Needed]
91
-
92
-
93
- #### Training Hyperparameters
94
-
95
- - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
-
97
- #### Speeds, Sizes, Times [optional]
98
-
99
- <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
-
101
- [More Information Needed]
102
-
103
- ## Evaluation
104
-
105
- <!-- This section describes the evaluation protocols and provides the results. -->
106
-
107
- ### Testing Data, Factors & Metrics
108
-
109
- #### Testing Data
110
-
111
- <!-- This should link to a Dataset Card if possible. -->
112
-
113
- [More Information Needed]
114
-
115
- #### Factors
116
-
117
- <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
-
119
- [More Information Needed]
120
-
121
- #### Metrics
122
-
123
- <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
-
125
- [More Information Needed]
126
-
127
- ### Results
128
-
129
- [More Information Needed]
130
-
131
- #### Summary
132
-
133
-
134
-
135
- ## Model Examination [optional]
136
-
137
- <!-- Relevant interpretability work for the model goes here -->
138
-
139
- [More Information Needed]
140
-
141
- ## Environmental Impact
142
-
143
- <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
-
145
- Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
146
-
147
- - **Hardware Type:** [More Information Needed]
148
- - **Hours used:** [More Information Needed]
149
- - **Cloud Provider:** [More Information Needed]
150
- - **Compute Region:** [More Information Needed]
151
- - **Carbon Emitted:** [More Information Needed]
152
-
153
- ## Technical Specifications [optional]
154
-
155
- ### Model Architecture and Objective
156
-
157
- [More Information Needed]
158
-
159
- ### Compute Infrastructure
160
-
161
- [More Information Needed]
162
-
163
- #### Hardware
164
-
165
- [More Information Needed]
166
-
167
- #### Software
168
-
169
- [More Information Needed]
170
-
171
- ## Citation [optional]
172
-
173
- <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
-
175
- **BibTeX:**
176
-
177
- [More Information Needed]
178
-
179
- **APA:**
180
-
181
- [More Information Needed]
182
-
183
- ## Glossary [optional]
184
-
185
- <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
-
187
- [More Information Needed]
188
-
189
- ## More Information [optional]
190
-
191
- [More Information Needed]
192
-
193
- ## Model Card Authors [optional]
194
-
195
- [More Information Needed]
196
-
197
- ## Model Card Contact
198
-
199
- [More Information Needed]
 
1
  ---
2
+ license: apache-2.0
3
  library_name: transformers
4
+ tags:
5
+ - automatic-speech-recognition
6
+ - smi
7
+ - sami
8
+ - pretraining
9
+ - continued pre-training
10
+ - CPT
11
  ---
12
+ # Sámi Wav2vec2-Large
13
 
14
+ The large model pre-trained on 16kHz sampled speech audio with **Continued Pre-Training (CPT)**. [wav2vec2-large-uralic-voxpopuli-v2](https://huggingface.co/facebook/wav2vec2-large-uralic-voxpopuli-v2) was used as a base model for continued pre-training. When using the model make sure that your speech input is also sampled at 16Khz.
15
 
16
+ **Note**: This model does not have a tokenizer as it was pre-trained on audio alone. In order to use this model **speech recognition**, a tokenizer should be created and the model should be fine-tuned on labeled text data. Check out [this blog](https://huggingface.co/blog/fine-tune-xlsr-wav2vec2) for a more in-depth explanation of how to fine-tune the model.
17
 
18
+ **Note**: Fine-tuned version is available at [GetmanY1/wav2vec2-large-sami-cont-pt-22k-finetuned](https://huggingface.co/GetmanY1/wav2vec2-large-sami-cont-pt-22k-finetuned)
19
 
20
+ ## Model description
21
 
22
+ The Sámi Wav2Vec2 Large has the same architecture and uses the same training objective as the English and multilingual one described in [Paper](https://arxiv.org/abs/2006.11477). It is pre-trained on 22.4k hours of unlabeled Sámi speech from [KAVI radio and television archive materials](https://kavi.fi/en/radio-ja-televisioarkistointia-vuodesta-2008/).
23
 
24
+ You can read more about the pre-trained model from [this paper](TODO).
25
 
26
+ ## Intended uses & limitations
27
 
28
+ You can use this model for Sámi ASR (speech-to-text) and SER (Spoken Emotion Recognition) tasks.
29
 
30
+ ### How to use
 
 
 
 
 
 
31
 
32
+ See [this notebook](https://colab.research.google.com/github/patrickvonplaten/notebooks/blob/master/Fine_Tune_XLS_R_on_Common_Voice.ipynb) for more information on how to fine-tune the model.
33
 
34
+ ### Limitations and bias
35
 
36
+ This model was pre-trained with audio samples whose maximum length was 30 seconds so this model most likely works the best for quite short audios of similar length. However, you can try this model with a lot longer audios too and see how it works. If you encounter out-of-memory errors with very long audio files you can use the audio chunking method introduced in [this blog post](https://huggingface.co/blog/asr-chunking).
 
 
37
 
38
+ The data used for pre-training was from the [KAVI](https://kavi.fi/en/radio-ja-televisioarkistointia-vuodesta-2008/) archives so this model might have biases towards the voices of radio hosts. The pre-training data was filtered via neural [VAD](https://huggingface.co/pyannote/voice-activity-detection), but some non-speech events like music might be still present in the training data, which might cause issues when fine-tuned on clear (no background noise) speech.
39
 
40
+ ## Training data
41
 
42
+ This model was pre-trained with 22.4k hours of Sámi speech data from the following sources:
43
 
44
+ | Dataset | Hours | % of total hours |
45
+ |:----------------------------------------------------------------------------------------------|:--------:|:----------------:|
46
+ | [YleSámiRadio](https://kavi.fi/en/radio-ja-televisioarkistointia-vuodesta-2008/) | 22415 h | 100 % |
47
 
 
48
 
49
+ Datasets were filtered to include a maximum length of 30 seconds long audio samples.
50
 
51
+ ## Training procedure
52
 
53
+ Training was done on 256 AMD MI250x GPU modules (512 GPUs from the software perspective), using [LUMI](https://www.lumi-supercomputer.eu/).
54
 
55
+ The training script was provided by Fairseq and it is available [here](https://github.com/facebookresearch/fairseq/tree/main/examples/wav2vec).
56
 
57
+ ### Training hyperparameters
58
 
59
+ The following hyperparameters were used during training:
60
+ - learning_rate: 2e-04
61
+ - max_update: 166667
62
+ - seed: 1
63
+ - optimizer: [8-bit Adam](https://github.com/facebookresearch/bitsandbytes) with betas=(0.9,0.98) and epsilon=1e-06
64
+ - lr_scheduler_type: linear
65
+ - lr_scheduler_warmup_updates: 5000
66
+ - fp16: true
67
+ - max_sample_size: 960000
68
+ - min_sample_size: 32000
69
+ - normalize: true
70
+ - max_tokens: 1800000
71
+ - distributed_world_size: 512
72
 
73
+ The pre-trained model was initialized with the following hyperparameters:
74
+ - quantize_targets: true
75
+ - latent_temp: [2.0, 0.5, 0.999995]
76
+ - extractor_mode: layer_norm
77
+ - layer_norm_first: true
78
+ - dropout_input: 0.0
79
+ - dropout_features: 0.0
80
+ - feature_grad_mult: 1.0
81
+ - encoder_embed_dim: 1024
82
+ - encoder_layers: 24
83
+ - encoder_ffn_embed_dim: 4096
84
+ - encoder_attention_heads: 16
85
+ - dropout: 0.0
86
+ - attention_dropout: 0.0
87
+ - activation_dropout: 0.0
88
+ - encoder_layerdrop: 0.0
89
 
90
+ ### Training results
91
 
92
+ | Training Loss | Epoch | Validation Loss |
93
+ |:-------------:|:------:|:---------------:|
94
+ | 2.821 | 1 | 2.735 |
95
+ | 2.25 | 25 | 2.157 |
96
+ | 2.161 | 50 | 2.063 |
97
+ | 2.102 | 75 | 2.04 |
98
+ | 2.07 | 97 | 2.011 |
99
 
 
100
 
101
+ ### Framework versions
102
 
103
+ - Pytorch 2.3.0+rocm6.0
104
+ - Fairseq 0.12.2
105
 
106
+ ## Team Members
107
 
108
+ - Yaroslav Getman, [Hugging Face profile](https://huggingface.co/GetmanY1), [LinkedIn profile](https://www.linkedin.com/in/yaroslav-getman/)
109
+ - Tamas Grosz, [Hugging Face profile](https://huggingface.co/Grosy), [LinkedIn profile](https://www.linkedin.com/in/tam%C3%A1s-gr%C3%B3sz-950a049a/)
110
 
111
+ Feel free to contact us for more details 🤗