gagan3012 commited on
Commit
6e2fe90
1 Parent(s): b7a5a8c

Initial commit

Browse files
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": 64,
74
+ "transformers_version": "4.4.0",
75
+ "vocab_size": 65
76
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:635fa64b04534d3fdbfeb0eb056f666a4dc3b69da6b382d04bdebd3c6de38551
3
+ size 2490610183
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:3c1ab6f51758354bce94b889b1117f9a4711fa401a7df791dd81f53366d065b0
3
+ size 1262200343
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87f9bbfc662da8015ef424c22600dffce065acb238f887c429fc16defbe58a59
3
+ size 623
trainer_state.json ADDED
@@ -0,0 +1,152 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.41098169717138106,
3
+ "best_model_checkpoint": "/content/wav2vec2-large-xlsr-nepali/checkpoint-2000",
4
+ "epoch": 34.47863247863248,
5
+ "global_step": 2000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 1.72,
12
+ "learning_rate": 0.00011999999999999999,
13
+ "loss": 9.4317,
14
+ "step": 100
15
+ },
16
+ {
17
+ "epoch": 3.44,
18
+ "learning_rate": 0.00023999999999999998,
19
+ "loss": 3.4467,
20
+ "step": 200
21
+ },
22
+ {
23
+ "epoch": 5.17,
24
+ "learning_rate": 0.0002943396226415094,
25
+ "loss": 3.3566,
26
+ "step": 300
27
+ },
28
+ {
29
+ "epoch": 6.89,
30
+ "learning_rate": 0.00028301886792452826,
31
+ "loss": 1.9291,
32
+ "step": 400
33
+ },
34
+ {
35
+ "epoch": 8.62,
36
+ "learning_rate": 0.00027169811320754714,
37
+ "loss": 0.7257,
38
+ "step": 500
39
+ },
40
+ {
41
+ "epoch": 10.34,
42
+ "learning_rate": 0.000260377358490566,
43
+ "loss": 0.4447,
44
+ "step": 600
45
+ },
46
+ {
47
+ "epoch": 12.07,
48
+ "learning_rate": 0.0002490566037735849,
49
+ "loss": 0.3317,
50
+ "step": 700
51
+ },
52
+ {
53
+ "epoch": 13.79,
54
+ "learning_rate": 0.00023773584905660377,
55
+ "loss": 0.2567,
56
+ "step": 800
57
+ },
58
+ {
59
+ "epoch": 15.51,
60
+ "learning_rate": 0.00022641509433962264,
61
+ "loss": 0.2293,
62
+ "step": 900
63
+ },
64
+ {
65
+ "epoch": 17.24,
66
+ "learning_rate": 0.00021509433962264146,
67
+ "loss": 0.1675,
68
+ "step": 1000
69
+ },
70
+ {
71
+ "epoch": 17.24,
72
+ "eval_loss": 0.3577611744403839,
73
+ "eval_runtime": 28.2181,
74
+ "eval_samples_per_second": 7.336,
75
+ "eval_wer": 0.44869661674986133,
76
+ "step": 1000
77
+ },
78
+ {
79
+ "epoch": 18.96,
80
+ "learning_rate": 0.00020377358490566034,
81
+ "loss": 0.1423,
82
+ "step": 1100
83
+ },
84
+ {
85
+ "epoch": 20.68,
86
+ "learning_rate": 0.00019245283018867922,
87
+ "loss": 0.1365,
88
+ "step": 1200
89
+ },
90
+ {
91
+ "epoch": 22.41,
92
+ "learning_rate": 0.0001811320754716981,
93
+ "loss": 0.1174,
94
+ "step": 1300
95
+ },
96
+ {
97
+ "epoch": 24.14,
98
+ "learning_rate": 0.00016981132075471697,
99
+ "loss": 0.1103,
100
+ "step": 1400
101
+ },
102
+ {
103
+ "epoch": 25.85,
104
+ "learning_rate": 0.00015849056603773584,
105
+ "loss": 0.0915,
106
+ "step": 1500
107
+ },
108
+ {
109
+ "epoch": 27.58,
110
+ "learning_rate": 0.0001471698113207547,
111
+ "loss": 0.0896,
112
+ "step": 1600
113
+ },
114
+ {
115
+ "epoch": 29.31,
116
+ "learning_rate": 0.00013584905660377357,
117
+ "loss": 0.0929,
118
+ "step": 1700
119
+ },
120
+ {
121
+ "epoch": 31.03,
122
+ "learning_rate": 0.00012452830188679244,
123
+ "loss": 0.075,
124
+ "step": 1800
125
+ },
126
+ {
127
+ "epoch": 32.75,
128
+ "learning_rate": 0.00011320754716981132,
129
+ "loss": 0.0745,
130
+ "step": 1900
131
+ },
132
+ {
133
+ "epoch": 34.48,
134
+ "learning_rate": 0.00010188679245283017,
135
+ "loss": 0.0603,
136
+ "step": 2000
137
+ },
138
+ {
139
+ "epoch": 34.48,
140
+ "eval_loss": 0.3987378180027008,
141
+ "eval_runtime": 27.73,
142
+ "eval_samples_per_second": 7.465,
143
+ "eval_wer": 0.41098169717138106,
144
+ "step": 2000
145
+ }
146
+ ],
147
+ "max_steps": 2900,
148
+ "num_train_epochs": 50,
149
+ "total_flos": 9.875307296349696e+18,
150
+ "trial_name": null,
151
+ "trial_params": null
152
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97af1403a94c748408eb00b9167fb9e8bade995019b4f065370a98b26f3a6c42
3
+ size 2287