Initial commit
Browse files- config.json +76 -0
- optimizer.pt +3 -0
- preprocessor_config.json +8 -0
- pytorch_model.bin +3 -0
- scheduler.pt +3 -0
- special_tokens_map.json +1 -0
- tokenizer_config.json +1 -0
- trainer_state.json +72 -0
- training_args.bin +3 -0
- vocab.json +1 -0
config.json
ADDED
@@ -0,0 +1,76 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "facebook/wav2vec2-large-xlsr-53",
|
3 |
+
"activation_dropout": 0.0,
|
4 |
+
"apply_spec_augment": true,
|
5 |
+
"architectures": [
|
6 |
+
"Wav2Vec2ForCTC"
|
7 |
+
],
|
8 |
+
"attention_dropout": 0.1,
|
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.0,
|
46 |
+
"final_dropout": 0.0,
|
47 |
+
"gradient_checkpointing": true,
|
48 |
+
"hidden_act": "gelu",
|
49 |
+
"hidden_dropout": 0.1,
|
50 |
+
"hidden_size": 1024,
|
51 |
+
"initializer_range": 0.02,
|
52 |
+
"intermediate_size": 4096,
|
53 |
+
"layer_norm_eps": 1e-05,
|
54 |
+
"layerdrop": 0.1,
|
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.05,
|
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": 72,
|
74 |
+
"transformers_version": "4.4.0",
|
75 |
+
"vocab_size": 73
|
76 |
+
}
|
optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:915f39749bdd822f47e68c0477d435a9fd7760c3d590bf63ba9e86e802c006e8
|
3 |
+
size 2490675719
|
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 |
+
}
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:835850a7a029e73a8fa6113cd193085d0e470700c4bb45be4e41f630bc8afe34
|
3 |
+
size 1262233111
|
scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:90a4b7dd943e4433a60f796edbc9ed88f16bd4c3bd4609e41aaaaf622ba391cb
|
3 |
+
size 623
|
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": "|"}
|
trainer_state.json
ADDED
@@ -0,0 +1,72 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 48.78048780487805,
|
5 |
+
"global_step": 4000,
|
6 |
+
"is_hyper_param_search": false,
|
7 |
+
"is_local_process_zero": true,
|
8 |
+
"is_world_process_zero": true,
|
9 |
+
"log_history": [
|
10 |
+
{
|
11 |
+
"epoch": 12.2,
|
12 |
+
"learning_rate": 0.0004200542005420054,
|
13 |
+
"loss": 2.2288,
|
14 |
+
"step": 1000
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"epoch": 12.2,
|
18 |
+
"eval_loss": 0.4119809865951538,
|
19 |
+
"eval_runtime": 39.8264,
|
20 |
+
"eval_samples_per_second": 7.307,
|
21 |
+
"eval_wer": 0.4587706146926537,
|
22 |
+
"step": 1000
|
23 |
+
},
|
24 |
+
{
|
25 |
+
"epoch": 24.39,
|
26 |
+
"learning_rate": 0.0002845528455284553,
|
27 |
+
"loss": 0.1307,
|
28 |
+
"step": 2000
|
29 |
+
},
|
30 |
+
{
|
31 |
+
"epoch": 24.39,
|
32 |
+
"eval_loss": 0.46079644560813904,
|
33 |
+
"eval_runtime": 39.8036,
|
34 |
+
"eval_samples_per_second": 7.311,
|
35 |
+
"eval_wer": 0.3800599700149925,
|
36 |
+
"step": 2000
|
37 |
+
},
|
38 |
+
{
|
39 |
+
"epoch": 36.59,
|
40 |
+
"learning_rate": 0.00014905149051490515,
|
41 |
+
"loss": 0.0645,
|
42 |
+
"step": 3000
|
43 |
+
},
|
44 |
+
{
|
45 |
+
"epoch": 36.59,
|
46 |
+
"eval_loss": 0.4456443190574646,
|
47 |
+
"eval_runtime": 41.0007,
|
48 |
+
"eval_samples_per_second": 7.097,
|
49 |
+
"eval_wer": 0.35007496251874065,
|
50 |
+
"step": 3000
|
51 |
+
},
|
52 |
+
{
|
53 |
+
"epoch": 48.78,
|
54 |
+
"learning_rate": 1.3550135501355015e-05,
|
55 |
+
"loss": 0.042,
|
56 |
+
"step": 4000
|
57 |
+
},
|
58 |
+
{
|
59 |
+
"epoch": 48.78,
|
60 |
+
"eval_loss": 0.46975016593933105,
|
61 |
+
"eval_runtime": 40.134,
|
62 |
+
"eval_samples_per_second": 7.251,
|
63 |
+
"eval_wer": 0.3365817091454273,
|
64 |
+
"step": 4000
|
65 |
+
}
|
66 |
+
],
|
67 |
+
"max_steps": 4100,
|
68 |
+
"num_train_epochs": 50,
|
69 |
+
"total_flos": 1.961718571271133e+19,
|
70 |
+
"trial_name": null,
|
71 |
+
"trial_params": null
|
72 |
+
}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b9eb3b3eafe5ebea11421306fb6d11ffc66297eddda9495cddcfd0a7770a2b8e
|
3 |
+
size 2287
|
vocab.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"แท": 0, "แ": 1, "แ": 2, "แ": 3, "แ": 4, "แ": 5, "แ": 6, "แ": 7, "แ": 8, "แ": 9, "แฑ": 10, "แง": 11, "แ": 12, "แ": 13, "แ": 14, "แ
": 15, "แ": 16, "แ": 17, "แ": 18, "แป": 19, "แ": 20, "แช": 21, "แ": 22, "แ": 23, "แฎ": 24, "แ ": 25, "แ": 26, "แ": 27, "แ": 28, "แ": 29, "แ": 30, "แญ": 31, "แ": 32, "แก": 33, "แ": 34, "แ": 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, "แผ": 64, "แน": 65, "แซ": 66, "แ": 67, "แ": 68, "แถ": 69, "แ": 70, "|": 35, "[UNK]": 71, "[PAD]": 72}
|