longdnk commited on
Commit
52efbd5
·
verified ·
1 Parent(s): 1b68ba3

Upload 10 files

Browse files
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d6b0967b5ab414903ec82cb4eb256dc82f207ea195239a335fbb7c0f1e434f7c
3
  size 377851056
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f5204db756e34f799719358f88672c8b4c6026f91eebda145d29f6b5a859364
3
  size 377851056
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ba2d5edcb13e7e96adbb0ca65cf4d6241e0e76c2d3114a8ecf24fc189d00c1bb
3
  size 377899102
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c63dd48407ceec67dee01431e0a481bb387803d9ef4f6a5541ecfead1b611e8
3
  size 377899102
rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d1e731ad63e753d284193b80f173a2ff1be45026693bc2996a76b495f4e45667
3
  size 14244
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3913ae9493cc4baad846c1b707d069986928d1245d876521c5d6546dda5464f
3
  size 14244
scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:da279824d75da67ee43e4d57e1cc5b4735e62b87cb30e4ce9cc7d11a432023a9
3
  size 1064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea297f7ab56af8b17d03b96627bb16862b6bf5fc7088f8cd1f0a0f4093d4464c
3
  size 1064
trainer_state.json CHANGED
@@ -1,198 +1,298 @@
1
  {
2
- "best_metric": 0.051292333751916885,
3
- "best_model_checkpoint": "./wav2vec2-base-demo/checkpoint-4500",
4
- "epoch": 10.0,
5
  "eval_steps": 500,
6
- "global_step": 4810,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
  "epoch": 1.0395010395010396,
13
- "grad_norm": 1.8444721698760986,
14
  "learning_rate": 4.99e-05,
15
- "loss": 0.1188,
16
  "step": 500
17
  },
18
  {
19
  "epoch": 1.0395010395010396,
20
- "eval_cer": 0.035194462663447716,
21
- "eval_loss": 0.06642936915159225,
22
- "eval_mer": 0.10481941437848924,
23
- "eval_runtime": 34.4969,
24
- "eval_samples_per_second": 44.555,
25
- "eval_steps_per_second": 5.595,
26
- "eval_wer": 0.10539580707984877,
27
- "eval_wil": 0.1892106514423464,
28
- "eval_wip": 0.8107893485576536,
29
  "step": 500
30
  },
31
  {
32
  "epoch": 2.079002079002079,
33
- "grad_norm": 1.2975250482559204,
34
- "learning_rate": 4.421113689095128e-05,
35
- "loss": 0.1151,
36
  "step": 1000
37
  },
38
  {
39
  "epoch": 2.079002079002079,
40
- "eval_cer": 0.034538578545611855,
41
- "eval_loss": 0.062206532806158066,
42
- "eval_mer": 0.10494778291388461,
43
- "eval_runtime": 34.1611,
44
- "eval_samples_per_second": 44.993,
45
- "eval_steps_per_second": 5.65,
46
- "eval_wer": 0.10533852674991408,
47
- "eval_wil": 0.19021462347071394,
48
- "eval_wip": 0.8097853765292861,
49
  "step": 1000
50
  },
51
  {
52
  "epoch": 3.1185031185031185,
53
- "grad_norm": 0.5322809815406799,
54
- "learning_rate": 3.8410672853828304e-05,
55
- "loss": 0.1235,
56
  "step": 1500
57
  },
58
  {
59
  "epoch": 3.1185031185031185,
60
- "eval_cer": 0.03377105457793159,
61
- "eval_loss": 0.062230244278907776,
62
- "eval_mer": 0.10482995141468991,
63
- "eval_runtime": 34.2894,
64
- "eval_samples_per_second": 44.824,
65
- "eval_steps_per_second": 5.629,
66
- "eval_wer": 0.10505212510024058,
67
- "eval_wil": 0.19100713135160063,
68
- "eval_wip": 0.8089928686483994,
69
  "step": 1500
70
  },
71
  {
72
  "epoch": 4.158004158004158,
73
- "grad_norm": 1.327663779258728,
74
- "learning_rate": 3.262180974477959e-05,
75
- "loss": 0.1068,
76
  "step": 2000
77
  },
78
  {
79
  "epoch": 4.158004158004158,
80
- "eval_cer": 0.03463626341422571,
81
- "eval_loss": 0.06272279471158981,
82
- "eval_mer": 0.10423917384606607,
83
- "eval_runtime": 34.2637,
84
- "eval_samples_per_second": 44.858,
85
- "eval_steps_per_second": 5.633,
86
- "eval_wer": 0.10465116279069768,
87
- "eval_wil": 0.18912080601859016,
88
- "eval_wip": 0.8108791939814098,
89
  "step": 2000
90
  },
91
  {
92
  "epoch": 5.197505197505198,
93
- "grad_norm": 1.7840451002120972,
94
- "learning_rate": 2.6821345707656613e-05,
95
- "loss": 0.1123,
96
  "step": 2500
97
  },
98
  {
99
  "epoch": 5.197505197505198,
100
- "eval_cer": 0.033366360122245635,
101
- "eval_loss": 0.06801871210336685,
102
- "eval_mer": 0.10333599908602764,
103
- "eval_runtime": 34.1023,
104
- "eval_samples_per_second": 45.07,
105
- "eval_steps_per_second": 5.659,
106
- "eval_wer": 0.10362011685187307,
107
- "eval_wil": 0.18807839647659041,
108
- "eval_wip": 0.8119216035234096,
109
  "step": 2500
110
  },
111
  {
112
  "epoch": 6.237006237006237,
113
- "grad_norm": 0.5479459762573242,
114
- "learning_rate": 2.1032482598607887e-05,
115
- "loss": 0.1066,
116
  "step": 3000
117
  },
118
  {
119
  "epoch": 6.237006237006237,
120
- "eval_cer": 0.03377105457793159,
121
- "eval_loss": 0.0692114531993866,
122
- "eval_mer": 0.10309631065746706,
123
- "eval_runtime": 34.3278,
124
- "eval_samples_per_second": 44.774,
125
- "eval_steps_per_second": 5.622,
126
- "eval_wer": 0.10356283652193837,
127
- "eval_wil": 0.18640279212307265,
128
- "eval_wip": 0.8135972078769274,
129
  "step": 3000
130
  },
131
  {
132
  "epoch": 7.276507276507276,
133
- "grad_norm": 2.529324769973755,
134
- "learning_rate": 1.5243619489559166e-05,
135
- "loss": 0.0985,
136
  "step": 3500
137
  },
138
  {
139
  "epoch": 7.276507276507276,
140
- "eval_cer": 0.032780250910562525,
141
- "eval_loss": 0.0544947050511837,
142
- "eval_mer": 0.1027412906910337,
143
- "eval_runtime": 34.2034,
144
- "eval_samples_per_second": 44.937,
145
- "eval_steps_per_second": 5.643,
146
- "eval_wer": 0.10304731355252607,
147
- "eval_wil": 0.1872840097439975,
148
- "eval_wip": 0.8127159902560025,
149
  "step": 3500
150
  },
151
  {
152
  "epoch": 8.316008316008316,
153
- "grad_norm": 2.9378857612609863,
154
- "learning_rate": 9.454756380510442e-06,
155
- "loss": 0.1145,
156
  "step": 4000
157
  },
158
  {
159
  "epoch": 8.316008316008316,
160
- "eval_cer": 0.03286398079794583,
161
- "eval_loss": 0.05142713338136673,
162
- "eval_mer": 0.10264465642314503,
163
- "eval_runtime": 34.3124,
164
- "eval_samples_per_second": 44.794,
165
- "eval_steps_per_second": 5.625,
166
- "eval_wer": 0.10293275289265666,
167
- "eval_wil": 0.18734075205582912,
168
- "eval_wip": 0.8126592479441709,
169
  "step": 4000
170
  },
171
  {
172
  "epoch": 9.355509355509355,
173
- "grad_norm": 1.8327735662460327,
174
- "learning_rate": 3.654292343387471e-06,
175
- "loss": 0.1128,
176
  "step": 4500
177
  },
178
  {
179
  "epoch": 9.355509355509355,
180
- "eval_cer": 0.03269652102317922,
181
- "eval_loss": 0.051292333751916885,
182
- "eval_mer": 0.10197669104204753,
183
- "eval_runtime": 34.3592,
184
- "eval_samples_per_second": 44.733,
185
- "eval_steps_per_second": 5.617,
186
- "eval_wer": 0.10224538893344026,
187
- "eval_wil": 0.1861754557735349,
188
- "eval_wip": 0.8138245442264651,
189
  "step": 4500
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
190
  }
191
  ],
192
  "logging_steps": 500,
193
- "max_steps": 4810,
194
  "num_input_tokens_seen": 0,
195
- "num_train_epochs": 10,
196
  "save_steps": 500,
197
  "stateful_callbacks": {
198
  "TrainerControl": {
@@ -201,12 +301,12 @@
201
  "should_evaluate": false,
202
  "should_log": false,
203
  "should_save": true,
204
- "should_training_stop": true
205
  },
206
  "attributes": {}
207
  }
208
  },
209
- "total_flos": 3.07474151587841e+18,
210
  "train_batch_size": 16,
211
  "trial_name": null,
212
  "trial_params": null
 
1
  {
2
+ "best_metric": 0.05392587557435036,
3
+ "best_model_checkpoint": "./wav2vec2-base-demo/checkpoint-7000",
4
+ "epoch": 14.553014553014552,
5
  "eval_steps": 500,
6
+ "global_step": 7000,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
  "epoch": 1.0395010395010396,
13
+ "grad_norm": 2.4276885986328125,
14
  "learning_rate": 4.99e-05,
15
+ "loss": 0.0935,
16
  "step": 500
17
  },
18
  {
19
  "epoch": 1.0395010395010396,
20
+ "eval_cer": 0.03704485933935964,
21
+ "eval_loss": 0.07600321620702744,
22
+ "eval_mer": 0.10765687967760507,
23
+ "eval_runtime": 34.3532,
24
+ "eval_samples_per_second": 44.741,
25
+ "eval_steps_per_second": 5.618,
26
+ "eval_wer": 0.1080737444373808,
27
+ "eval_wil": 0.19400502379268658,
28
+ "eval_wip": 0.8059949762073134,
29
  "step": 500
30
  },
31
  {
32
  "epoch": 2.079002079002079,
33
+ "grad_norm": 1.2353624105453491,
34
+ "learning_rate": 4.8671291355389544e-05,
35
+ "loss": 0.0914,
36
  "step": 1000
37
  },
38
  {
39
  "epoch": 2.079002079002079,
40
+ "eval_cer": 0.037467263664779926,
41
+ "eval_loss": 0.08513512462377548,
42
+ "eval_mer": 0.1085740644640489,
43
+ "eval_runtime": 34.2139,
44
+ "eval_samples_per_second": 44.923,
45
+ "eval_steps_per_second": 5.641,
46
+ "eval_wer": 0.10882505923828238,
47
+ "eval_wil": 0.19587239193616368,
48
+ "eval_wip": 0.8041276080638363,
49
  "step": 1000
50
  },
51
  {
52
  "epoch": 3.1185031185031185,
53
+ "grad_norm": 0.4545057415962219,
54
+ "learning_rate": 4.733724653148346e-05,
55
+ "loss": 0.1007,
56
  "step": 1500
57
  },
58
  {
59
  "epoch": 3.1185031185031185,
60
+ "eval_cer": 0.03667877559066205,
61
+ "eval_loss": 0.08273813873529434,
62
+ "eval_mer": 0.10847477196628565,
63
+ "eval_runtime": 34.1503,
64
+ "eval_samples_per_second": 45.007,
65
+ "eval_steps_per_second": 5.651,
66
+ "eval_wer": 0.10859388545338959,
67
+ "eval_wil": 0.1960486347087158,
68
+ "eval_wip": 0.8039513652912842,
69
  "step": 1500
70
  },
71
  {
72
  "epoch": 4.158004158004158,
73
+ "grad_norm": 0.7860220074653625,
74
+ "learning_rate": 4.600586979722519e-05,
75
+ "loss": 0.0883,
76
  "step": 2000
77
  },
78
  {
79
  "epoch": 4.158004158004158,
80
+ "eval_cer": 0.03734054236715384,
81
+ "eval_loss": 0.08130443096160889,
82
+ "eval_mer": 0.10836359444348377,
83
+ "eval_runtime": 34.1136,
84
+ "eval_samples_per_second": 45.055,
85
+ "eval_steps_per_second": 5.658,
86
+ "eval_wer": 0.10865167889961279,
87
+ "eval_wil": 0.19535745081720401,
88
+ "eval_wip": 0.804642549182796,
89
  "step": 2000
90
  },
91
  {
92
  "epoch": 5.197505197505198,
93
+ "grad_norm": 5.241793632507324,
94
+ "learning_rate": 4.46718249733191e-05,
95
+ "loss": 0.0988,
96
  "step": 2500
97
  },
98
  {
99
  "epoch": 5.197505197505198,
100
+ "eval_cer": 0.03660837486975867,
101
+ "eval_loss": 0.07141255587339401,
102
+ "eval_mer": 0.10702128887093983,
103
+ "eval_runtime": 34.0472,
104
+ "eval_samples_per_second": 45.143,
105
+ "eval_steps_per_second": 5.669,
106
+ "eval_wer": 0.10720684274403283,
107
+ "eval_wil": 0.1936232791597472,
108
+ "eval_wip": 0.8063767208402528,
109
  "step": 2500
110
  },
111
  {
112
  "epoch": 6.237006237006237,
113
+ "grad_norm": 1.3025474548339844,
114
+ "learning_rate": 4.334044823906083e-05,
115
+ "loss": 0.0962,
116
  "step": 3000
117
  },
118
  {
119
  "epoch": 6.237006237006237,
120
+ "eval_cer": 0.035636844921291995,
121
+ "eval_loss": 0.06766052544116974,
122
+ "eval_mer": 0.1056455798396863,
123
+ "eval_runtime": 34.1329,
124
+ "eval_samples_per_second": 45.03,
125
+ "eval_steps_per_second": 5.654,
126
+ "eval_wer": 0.10587759348089927,
127
+ "eval_wil": 0.1915670539174873,
128
+ "eval_wip": 0.8084329460825127,
129
  "step": 3000
130
  },
131
  {
132
  "epoch": 7.276507276507276,
133
+ "grad_norm": 1.1820029020309448,
134
+ "learning_rate": 4.200907150480256e-05,
135
+ "loss": 0.0951,
136
  "step": 3500
137
  },
138
  {
139
  "epoch": 7.276507276507276,
140
+ "eval_cer": 0.03551012362366591,
141
+ "eval_loss": 0.06142089143395424,
142
+ "eval_mer": 0.10593122547888299,
143
+ "eval_runtime": 34.0865,
144
+ "eval_samples_per_second": 45.091,
145
+ "eval_steps_per_second": 5.662,
146
+ "eval_wer": 0.10610876726579206,
147
+ "eval_wil": 0.19198195094690818,
148
+ "eval_wip": 0.8080180490530918,
149
  "step": 3500
150
  },
151
  {
152
  "epoch": 8.316008316008316,
153
+ "grad_norm": 3.633211851119995,
154
+ "learning_rate": 4.0677694770544295e-05,
155
+ "loss": 0.1161,
156
  "step": 4000
157
  },
158
  {
159
  "epoch": 8.316008316008316,
160
+ "eval_cer": 0.03551012362366591,
161
+ "eval_loss": 0.06067756190896034,
162
+ "eval_mer": 0.10610691424946658,
163
+ "eval_runtime": 34.0596,
164
+ "eval_samples_per_second": 45.127,
165
+ "eval_steps_per_second": 5.667,
166
+ "eval_wer": 0.10633994105068485,
167
+ "eval_wil": 0.19244783032392676,
168
+ "eval_wip": 0.8075521696760732,
169
  "step": 4000
170
  },
171
  {
172
  "epoch": 9.355509355509355,
173
+ "grad_norm": 1.8992377519607544,
174
+ "learning_rate": 3.934364994663821e-05,
175
+ "loss": 0.1202,
176
  "step": 4500
177
  },
178
  {
179
  "epoch": 9.355509355509355,
180
+ "eval_cer": 0.036453493283771225,
181
+ "eval_loss": 0.07183138281106949,
182
+ "eval_mer": 0.10711813567143516,
183
+ "eval_runtime": 34.0394,
184
+ "eval_samples_per_second": 45.154,
185
+ "eval_steps_per_second": 5.67,
186
+ "eval_wer": 0.10732242963647923,
187
+ "eval_wil": 0.19300207421517157,
188
+ "eval_wip": 0.8069979257848284,
189
  "step": 4500
190
+ },
191
+ {
192
+ "epoch": 10.395010395010395,
193
+ "grad_norm": 1.5443813800811768,
194
+ "learning_rate": 3.800960512273213e-05,
195
+ "loss": 0.1431,
196
+ "step": 5000
197
+ },
198
+ {
199
+ "epoch": 10.395010395010395,
200
+ "eval_cer": 0.036031088958350936,
201
+ "eval_loss": 0.0653519257903099,
202
+ "eval_mer": 0.10687726984492996,
203
+ "eval_runtime": 34.4227,
204
+ "eval_samples_per_second": 44.651,
205
+ "eval_steps_per_second": 5.607,
206
+ "eval_wer": 0.10714904929780963,
207
+ "eval_wil": 0.19299958643561255,
208
+ "eval_wip": 0.8070004135643875,
209
+ "step": 5000
210
+ },
211
+ {
212
+ "epoch": 11.434511434511435,
213
+ "grad_norm": 0.9764054417610168,
214
+ "learning_rate": 3.667556029882604e-05,
215
+ "loss": 0.1356,
216
+ "step": 5500
217
+ },
218
+ {
219
+ "epoch": 11.434511434511435,
220
+ "eval_cer": 0.03546788319112388,
221
+ "eval_loss": 0.07734616100788116,
222
+ "eval_mer": 0.10516307479543621,
223
+ "eval_runtime": 34.1476,
224
+ "eval_samples_per_second": 45.01,
225
+ "eval_steps_per_second": 5.652,
226
+ "eval_wer": 0.10547303935733687,
227
+ "eval_wil": 0.18957490448529868,
228
+ "eval_wip": 0.8104250955147013,
229
+ "step": 5500
230
+ },
231
+ {
232
+ "epoch": 12.474012474012474,
233
+ "grad_norm": 1.104174017906189,
234
+ "learning_rate": 3.5341515474919964e-05,
235
+ "loss": 0.1561,
236
+ "step": 6000
237
+ },
238
+ {
239
+ "epoch": 12.474012474012474,
240
+ "eval_cer": 0.035862127228182814,
241
+ "eval_loss": 0.06988665461540222,
242
+ "eval_mer": 0.10506015773415463,
243
+ "eval_runtime": 34.0048,
244
+ "eval_samples_per_second": 45.2,
245
+ "eval_steps_per_second": 5.676,
246
+ "eval_wer": 0.10547303935733687,
247
+ "eval_wil": 0.1890727356961288,
248
+ "eval_wip": 0.8109272643038712,
249
+ "step": 6000
250
+ },
251
+ {
252
+ "epoch": 13.513513513513514,
253
+ "grad_norm": 4.416908264160156,
254
+ "learning_rate": 3.400747065101387e-05,
255
+ "loss": 0.134,
256
+ "step": 6500
257
+ },
258
+ {
259
+ "epoch": 13.513513513513514,
260
+ "eval_cer": 0.0349750781448002,
261
+ "eval_loss": 0.0634101927280426,
262
+ "eval_mer": 0.10508709193678625,
263
+ "eval_runtime": 34.1003,
264
+ "eval_samples_per_second": 45.073,
265
+ "eval_steps_per_second": 5.66,
266
+ "eval_wer": 0.10529965901866728,
267
+ "eval_wil": 0.19050757507724403,
268
+ "eval_wip": 0.809492424922756,
269
+ "step": 6500
270
+ },
271
+ {
272
+ "epoch": 14.553014553014552,
273
+ "grad_norm": 0.7748957276344299,
274
+ "learning_rate": 3.267342582710779e-05,
275
+ "loss": 0.1304,
276
+ "step": 7000
277
+ },
278
+ {
279
+ "epoch": 14.553014553014552,
280
+ "eval_cer": 0.03490467742389682,
281
+ "eval_loss": 0.05392587557435036,
282
+ "eval_mer": 0.10498097544102386,
283
+ "eval_runtime": 34.1635,
284
+ "eval_samples_per_second": 44.99,
285
+ "eval_steps_per_second": 5.649,
286
+ "eval_wer": 0.10524186557244408,
287
+ "eval_wil": 0.18975677659084123,
288
+ "eval_wip": 0.8102432234091588,
289
+ "step": 7000
290
  }
291
  ],
292
  "logging_steps": 500,
293
+ "max_steps": 19240,
294
  "num_input_tokens_seen": 0,
295
+ "num_train_epochs": 40,
296
  "save_steps": 500,
297
  "stateful_callbacks": {
298
  "TrainerControl": {
 
301
  "should_evaluate": false,
302
  "should_log": false,
303
  "should_save": true,
304
+ "should_training_stop": false
305
  },
306
  "attributes": {}
307
  }
308
  },
309
+ "total_flos": 4.4824764428392996e+18,
310
  "train_batch_size": 16,
311
  "trial_name": null,
312
  "trial_params": null
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:adb2dde227c901afcc4b8ad055a16de87aa03de53761c737e130e66fa8b7dd9a
3
  size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:775696d645371d4025f4adc46331697961085adc3d0812695e8af16d5dc92a47
3
  size 5176