waveletdeboshir commited on
Commit
762fa07
1 Parent(s): f1d2f3f

Add info about model

Browse files
Files changed (1) hide show
  1. README.md +38 -3
README.md CHANGED
@@ -1,3 +1,38 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - ru
5
+ library_name: transformers
6
+ pipeline_tag: automatic-speech-recognition
7
+ tags:
8
+ - asr
9
+ - Pytorch
10
+ - pruned
11
+ - audio
12
+ - automatic-speech-recognition
13
+ ---
14
+
15
+ # Whisper-small-ru-pruned
16
+
17
+ ## Model info
18
+ This is a pruned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) model with only russian tokens left.
19
+ Pruning was made without any fine-tuning. Method from [this post](https://medium.com/m/global-identity-2?redirectUrl=https%3A%2F%2Ftowardsdatascience.com%2Fhow-to-adapt-a-multilingual-t5-model-for-a-single-language-b9f94f3d9c90) was used.
20
+
21
+ ## Size
22
+ Only 10% tokens was left including special whisper tokens, added whisper tokens, 100 most popular tokens from tokenizer and 3000 most popular Russian tokens computed by tokenization of russian text corpus.
23
+
24
+ Model size is 15% less then original whisper-small:
25
+ | | openai/whisper-small | waveletdeboshir/whisper-small-ru-pruned |
26
+ | :------ | :------ | :------ |
27
+ | n of parameters | 242 M | 205 M |
28
+ | n of parameters (with proj_out layer) | 281 M | 209 M |
29
+ | model file size | 967 Mb | 837 Mb |
30
+ | vocab_size | 51865 | 4705 |
31
+
32
+ ## Metrics
33
+ TODO
34
+
35
+ You can fine-tune this model on your data to achive better performance.
36
+
37
+ ## Colab for pruning
38
+ TODO