basilkr commited on
Commit
aff3263
·
1 Parent(s): 1448665

Upload 9 files

Browse files
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "openai/whisper-small",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "gelu",
5
+ "architectures": [
6
+ "WhisperForConditionalGeneration"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "begin_suppress_tokens": [
10
+ 220,
11
+ 50257
12
+ ],
13
+ "bos_token_id": 50257,
14
+ "d_model": 768,
15
+ "decoder_attention_heads": 12,
16
+ "decoder_ffn_dim": 3072,
17
+ "decoder_layerdrop": 0.0,
18
+ "decoder_layers": 12,
19
+ "decoder_start_token_id": 50258,
20
+ "dropout": 0.0,
21
+ "encoder_attention_heads": 12,
22
+ "encoder_ffn_dim": 3072,
23
+ "encoder_layerdrop": 0.0,
24
+ "encoder_layers": 12,
25
+ "eos_token_id": 50257,
26
+ "forced_decoder_ids": null,
27
+ "init_std": 0.02,
28
+ "is_encoder_decoder": true,
29
+ "max_length": 500,
30
+ "max_source_positions": 1500,
31
+ "max_target_positions": 448,
32
+ "model_type": "whisper",
33
+ "num_hidden_layers": 12,
34
+ "num_mel_bins": 80,
35
+ "pad_token_id": 50257,
36
+ "scale_embedding": false,
37
+ "suppress_tokens": [],
38
+ "torch_dtype": "float32",
39
+ "transformers_version": "4.26.0",
40
+ "use_cache": true,
41
+ "vocab_size": 51865
42
+ }
generation_config.json ADDED
@@ -0,0 +1,112 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "begin_suppress_tokens": [
3
+ 220,
4
+ 50257
5
+ ],
6
+ "bos_token_id": 50257,
7
+ "decoder_start_token_id": 50258,
8
+ "eos_token_id": 50257,
9
+ "forced_decoder_ids": [
10
+ [
11
+ 1,
12
+ null
13
+ ],
14
+ [
15
+ 2,
16
+ 50359
17
+ ]
18
+ ],
19
+ "max_length": 448,
20
+ "pad_token_id": 50257,
21
+ "suppress_tokens": [
22
+ 1,
23
+ 2,
24
+ 7,
25
+ 8,
26
+ 9,
27
+ 10,
28
+ 14,
29
+ 25,
30
+ 26,
31
+ 27,
32
+ 28,
33
+ 29,
34
+ 31,
35
+ 58,
36
+ 59,
37
+ 60,
38
+ 61,
39
+ 62,
40
+ 63,
41
+ 90,
42
+ 91,
43
+ 92,
44
+ 93,
45
+ 359,
46
+ 503,
47
+ 522,
48
+ 542,
49
+ 873,
50
+ 893,
51
+ 902,
52
+ 918,
53
+ 922,
54
+ 931,
55
+ 1350,
56
+ 1853,
57
+ 1982,
58
+ 2460,
59
+ 2627,
60
+ 3246,
61
+ 3253,
62
+ 3268,
63
+ 3536,
64
+ 3846,
65
+ 3961,
66
+ 4183,
67
+ 4667,
68
+ 6585,
69
+ 6647,
70
+ 7273,
71
+ 9061,
72
+ 9383,
73
+ 10428,
74
+ 10929,
75
+ 11938,
76
+ 12033,
77
+ 12331,
78
+ 12562,
79
+ 13793,
80
+ 14157,
81
+ 14635,
82
+ 15265,
83
+ 15618,
84
+ 16553,
85
+ 16604,
86
+ 18362,
87
+ 18956,
88
+ 20075,
89
+ 21675,
90
+ 22520,
91
+ 26130,
92
+ 26161,
93
+ 26435,
94
+ 28279,
95
+ 29464,
96
+ 31650,
97
+ 32302,
98
+ 32470,
99
+ 36865,
100
+ 42863,
101
+ 47425,
102
+ 49870,
103
+ 50254,
104
+ 50258,
105
+ 50358,
106
+ 50359,
107
+ 50360,
108
+ 50361,
109
+ 50362
110
+ ],
111
+ "transformers_version": "4.26.0"
112
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5f45693f581d2fa3c7cf53d048fa2d992f41ef81ee6ad4ea5e7ebec8848142d
3
+ size 1934147269
preprocessor_config.json ADDED
The diff for this file is too large to render. See raw diff
 
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:682e8d4ee36b72f1b30932a5ca5c24a3f5d7ab7e5b3193b0b15f16229cfa4732
3
+ size 967095881
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb4efdc45f0d3e9562a6a346b6d3e00be0c5acde5e75f4745a77600d4c9748c5
3
+ size 13553
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d20d7e0a6973340b1b7ca962838e135530980f0c06cd48538b3fcece2c4acd05
3
+ size 627
trainer_state.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 97.17514124293785,
3
+ "best_model_checkpoint": "./whisper-small-hi\\checkpoint-30",
4
+ "epoch": 30.0,
5
+ "global_step": 30,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 10.0,
12
+ "eval_loss": 2.1109774112701416,
13
+ "eval_runtime": 161.1687,
14
+ "eval_samples_per_second": 0.068,
15
+ "eval_steps_per_second": 0.012,
16
+ "eval_wer": 101.69491525423729,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 20.0,
21
+ "eval_loss": 1.7198082208633423,
22
+ "eval_runtime": 158.8454,
23
+ "eval_samples_per_second": 0.069,
24
+ "eval_steps_per_second": 0.013,
25
+ "eval_wer": 100.0,
26
+ "step": 20
27
+ },
28
+ {
29
+ "epoch": 25.0,
30
+ "learning_rate": 5.000000000000001e-07,
31
+ "loss": 2.007,
32
+ "step": 25
33
+ },
34
+ {
35
+ "epoch": 30.0,
36
+ "eval_loss": 1.3272284269332886,
37
+ "eval_runtime": 185.4253,
38
+ "eval_samples_per_second": 0.059,
39
+ "eval_steps_per_second": 0.011,
40
+ "eval_wer": 97.17514124293785,
41
+ "step": 30
42
+ }
43
+ ],
44
+ "max_steps": 40,
45
+ "num_train_epochs": 40,
46
+ "total_flos": 9.52331821056e+16,
47
+ "trial_name": null,
48
+ "trial_params": null
49
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:650c4915923bf420700799c827055a5f8166bb2b78f03ce97a397efd4c0d503b
3
+ size 3579