speech-test commited on
Commit
2f39c97
1 Parent(s): 59435be

Upload model

Browse files
Files changed (4) hide show
  1. README.md +27 -0
  2. config.json +84 -0
  3. preprocessor_config.json +9 -0
  4. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ datasets:
4
+ - librispeech_asr
5
+ tags:
6
+ - speech
7
+ license: apache-2.0
8
+ ---
9
+
10
+ # SEW-mid
11
+
12
+ [SEW by ASAPP Research](https://github.com/asappresearch/sew)
13
+
14
+ The base model pretrained on 16kHz sampled speech audio. When using the model make sure that your speech input is also sampled at 16Khz. Note that this model should be fine-tuned on a downstream task, like Automatic Speech Recognition, Speaker Identification, Intent Classification, Emotion Recognition, etc...
15
+
16
+ Paper: [Performance-Efficiency Trade-offs in Unsupervised Pre-training for Speech Recognition](https://arxiv.org/abs/2109.06870)
17
+
18
+ Authors: Felix Wu, Kwangyoun Kim, Jing Pan, Kyu Han, Kilian Q. Weinberger, Yoav Artzi
19
+
20
+ **Abstract**
21
+ This paper is a study of performance-efficiency trade-offs in pre-trained models for automatic speech recognition (ASR). We focus on wav2vec 2.0, and formalize several architecture designs that influence both the model performance and its efficiency. Putting together all our observations, we introduce SEW (Squeezed and Efficient Wav2vec), a pre-trained model architecture with significant improvements along both performance and efficiency dimensions across a variety of training setups. For example, under the 100h-960h semi-supervised setup on LibriSpeech, SEW achieves a 1.9x inference speedup compared to wav2vec 2.0, with a 13.5% relative reduction in word error rate. With a similar inference time, SEW reduces word error rate by 25-50% across different model sizes.
22
+
23
+ The original model can be found under https://github.com/asappresearch/sew#model-checkpoints .
24
+
25
+ # Usage
26
+
27
+ See [this blog](https://huggingface.co/blog/fine-tune-wav2vec2-english) for more information on how to fine-tune the model. Note that the class `Wav2Vec2ForCTC` has to be replaced by `SEWForCTC`.
config.json ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_dropout": 0.0,
3
+ "apply_spec_augment": true,
4
+ "architectures": [
5
+ "SEWModel"
6
+ ],
7
+ "attention_dropout": 0.1,
8
+ "bos_token_id": 1,
9
+ "conv_bias": false,
10
+ "conv_dim": [
11
+ 64,
12
+ 128,
13
+ 128,
14
+ 128,
15
+ 128,
16
+ 256,
17
+ 256,
18
+ 256,
19
+ 256,
20
+ 512,
21
+ 512,
22
+ 512,
23
+ 512
24
+ ],
25
+ "conv_kernel": [
26
+ 10,
27
+ 3,
28
+ 1,
29
+ 3,
30
+ 1,
31
+ 3,
32
+ 1,
33
+ 3,
34
+ 1,
35
+ 2,
36
+ 1,
37
+ 2,
38
+ 1
39
+ ],
40
+ "conv_stride": [
41
+ 5,
42
+ 2,
43
+ 1,
44
+ 2,
45
+ 1,
46
+ 2,
47
+ 1,
48
+ 2,
49
+ 1,
50
+ 2,
51
+ 1,
52
+ 2,
53
+ 1
54
+ ],
55
+ "ctc_loss_reduction": "sum",
56
+ "ctc_zero_infinity": false,
57
+ "eos_token_id": 2,
58
+ "feat_extract_activation": "gelu",
59
+ "feat_extract_norm": "group",
60
+ "feat_proj_dropout": 0.1,
61
+ "final_dropout": 0.0,
62
+ "hidden_act": "gelu",
63
+ "hidden_dropout": 0.1,
64
+ "hidden_size": 768,
65
+ "initializer_range": 0.02,
66
+ "intermediate_size": 3072,
67
+ "layer_norm_eps": 1e-05,
68
+ "layerdrop": 0.2,
69
+ "mask_feature_length": 10,
70
+ "mask_feature_prob": 0.0,
71
+ "mask_time_length": 10,
72
+ "mask_time_prob": 0.65,
73
+ "model_type": "sew",
74
+ "num_attention_heads": 12,
75
+ "num_conv_pos_embedding_groups": 16,
76
+ "num_conv_pos_embeddings": 31,
77
+ "num_feat_extract_layers": 13,
78
+ "num_hidden_layers": 24,
79
+ "pad_token_id": 0,
80
+ "squeeze_factor": 2,
81
+ "torch_dtype": "float32",
82
+ "transformers_version": "4.12.0.dev0",
83
+ "vocab_size": 32
84
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "feature_extractor_type": "Wav2Vec2FeatureExtractor",
4
+ "feature_size": 1,
5
+ "padding_side": "right",
6
+ "padding_value": 0,
7
+ "return_attention_mask": false,
8
+ "sampling_rate": 16000
9
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3afa957e910a79932e836b9bf3f910a3029dfc678e420cec4578147b930540e
3
+ size 698851723