kmfoda commited on
Commit
30e4171
1 Parent(s): 73273e8

First version of the wav2vec2-large-xlsr-arabice model and tokenizer.

Browse files
.gitattributes CHANGED
@@ -13,4 +13,4 @@
13
  *.msgpack filter=lfs diff=lfs merge=lfs -text
14
  *.pb filter=lfs diff=lfs merge=lfs -text
15
  *.pt filter=lfs diff=lfs merge=lfs -text
16
- *.pth filter=lfs diff=lfs merge=lfs -text
 
13
  *.msgpack filter=lfs diff=lfs merge=lfs -text
14
  *.pb filter=lfs diff=lfs merge=lfs -text
15
  *.pt filter=lfs diff=lfs merge=lfs -text
16
+ *.pth filter=lfs diff=lfs merge=lfs -text
all_results.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "eval_loss": 1.1619502305984497,
4
+ "eval_mem_cpu_alloc_delta": 1765203650,
5
+ "eval_mem_cpu_peaked_delta": 70416270,
6
+ "eval_mem_gpu_alloc_delta": 0,
7
+ "eval_mem_gpu_peaked_delta": 5896910848,
8
+ "eval_runtime": 883.3879,
9
+ "eval_samples": 7622,
10
+ "eval_samples_per_second": 8.628,
11
+ "eval_wer": 0.6438685083419972,
12
+ "init_mem_cpu_alloc_delta": 620838,
13
+ "init_mem_cpu_peaked_delta": 18306,
14
+ "init_mem_gpu_alloc_delta": 1262070784,
15
+ "init_mem_gpu_peaked_delta": 0,
16
+ "train_mem_cpu_alloc_delta": 0,
17
+ "train_mem_cpu_peaked_delta": 0,
18
+ "train_mem_gpu_alloc_delta": 3788204544,
19
+ "train_mem_gpu_peaked_delta": 0,
20
+ "train_runtime": 115603.8298,
21
+ "train_samples": 21744,
22
+ "train_samples_per_second": 0.176
23
+ }
config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/wav2vec2-large-xlsr-53",
3
+ "activation_dropout": 0.055,
4
+ "apply_spec_augment": true,
5
+ "architectures": [
6
+ "Wav2Vec2ForCTC"
7
+ ],
8
+ "attention_dropout": 0.094,
9
+ "bos_token_id": 1,
10
+ "conv_bias": true,
11
+ "conv_dim": [
12
+ 512,
13
+ 512,
14
+ 512,
15
+ 512,
16
+ 512,
17
+ 512,
18
+ 512
19
+ ],
20
+ "conv_kernel": [
21
+ 10,
22
+ 3,
23
+ 3,
24
+ 3,
25
+ 3,
26
+ 2,
27
+ 2
28
+ ],
29
+ "conv_stride": [
30
+ 5,
31
+ 2,
32
+ 2,
33
+ 2,
34
+ 2,
35
+ 2,
36
+ 2
37
+ ],
38
+ "ctc_loss_reduction": "mean",
39
+ "ctc_zero_infinity": false,
40
+ "do_stable_layer_norm": true,
41
+ "eos_token_id": 2,
42
+ "feat_extract_activation": "gelu",
43
+ "feat_extract_dropout": 0.0,
44
+ "feat_extract_norm": "layer",
45
+ "feat_proj_dropout": 0.04,
46
+ "final_dropout": 0.0,
47
+ "gradient_checkpointing": true,
48
+ "hidden_act": "gelu",
49
+ "hidden_dropout": 0.047,
50
+ "hidden_size": 1024,
51
+ "initializer_range": 0.02,
52
+ "intermediate_size": 4096,
53
+ "layer_norm_eps": 1e-05,
54
+ "layerdrop": 0.041,
55
+ "mask_channel_length": 10,
56
+ "mask_channel_min_space": 1,
57
+ "mask_channel_other": 0.0,
58
+ "mask_channel_prob": 0.0,
59
+ "mask_channel_selection": "static",
60
+ "mask_feature_length": 10,
61
+ "mask_feature_prob": 0.0,
62
+ "mask_time_length": 10,
63
+ "mask_time_min_space": 1,
64
+ "mask_time_other": 0.0,
65
+ "mask_time_prob": 0.082,
66
+ "mask_time_selection": "static",
67
+ "model_type": "wav2vec2",
68
+ "num_attention_heads": 16,
69
+ "num_conv_pos_embedding_groups": 16,
70
+ "num_conv_pos_embeddings": 128,
71
+ "num_feat_extract_layers": 7,
72
+ "num_hidden_layers": 24,
73
+ "pad_token_id": 76,
74
+ "transformers_version": "4.5.0.dev0",
75
+ "vocab_size": 77
76
+ }
eval_results.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "eval_loss": 1.1619502305984497,
4
+ "eval_mem_cpu_alloc_delta": 1765203650,
5
+ "eval_mem_cpu_peaked_delta": 70416270,
6
+ "eval_mem_gpu_alloc_delta": 0,
7
+ "eval_mem_gpu_peaked_delta": 5896910848,
8
+ "eval_runtime": 883.3879,
9
+ "eval_samples": 7622,
10
+ "eval_samples_per_second": 8.628,
11
+ "eval_wer": 0.6438685083419972
12
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "feature_size": 1,
4
+ "padding_side": "right",
5
+ "padding_value": 0.0,
6
+ "return_attention_mask": true,
7
+ "sampling_rate": 16000
8
+ }
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"bos_token": "<s>", "eos_token": "</s>", "unk_token": "[UNK]", "pad_token": "[PAD]"}
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"unk_token": "[UNK]", "bos_token": "<s>", "eos_token": "</s>", "pad_token": "[PAD]", "do_lower_case": false, "word_delimiter_token": "|"}
train_results.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "init_mem_cpu_alloc_delta": 620838,
4
+ "init_mem_cpu_peaked_delta": 18306,
5
+ "init_mem_gpu_alloc_delta": 1262070784,
6
+ "init_mem_gpu_peaked_delta": 0,
7
+ "train_mem_cpu_alloc_delta": 0,
8
+ "train_mem_cpu_peaked_delta": 0,
9
+ "train_mem_gpu_alloc_delta": 3788204544,
10
+ "train_mem_gpu_peaked_delta": 0,
11
+ "train_runtime": 115603.8298,
12
+ "train_samples": 21744,
13
+ "train_samples_per_second": 0.176
14
+ }
vocab.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"t": 0, "ه": 1, "ح": 2, "؛": 3, "ى": 4, "ف": 5, "غ": 6, "ج": 7, "ع": 8, "چ": 9, ".": 10, "ﻻ": 11, "ٌ": 12, "e": 13, "ْ": 14, ":": 15, "،": 16, "خ": 17, "ئ": 18, "َ": 19, "ﺃ": 20, "ب": 21, "ڨ": 22, "أ": 23, "د": 24, "ا": 25, "ً": 26, ";": 27, "ذ": 28, "و": 29, "ة": 30, "-": 31, "_": 32, "ّ": 34, "\"": 35, "?": 36, "آ": 37, "ء": 38, "—": 39, "”": 40, "ش": 41, "“": 42, "ر": 43, "ٍ": 44, "ـ": 45, "ك": 46, "!": 47, "ي": 48, ",": 49, "ط": 50, "ھ": 51, "ز": 52, "م": 53, "ض": 54, "ِ": 55, "ؤ": 56, "ث": 57, "ُ": 58, "ق": 59, "ٰ": 60, "؟": 61, "ص": 62, "ۚ": 63, "g": 64, "ۖ": 65, "☭": 66, "ل": 67, "ی": 68, "ن": 69, "س": 70, "ظ": 71, "ک": 72, "ت": 73, "إ": 74, "|": 33, "[UNK]": 75, "[PAD]": 76}