Mamadou-Aw commited on
Commit
f87929b
·
verified ·
1 Parent(s): 40c5332

Upload folder using huggingface_hub

Browse files
config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_dropout": 0.0,
3
+ "activation_function": "gelu",
4
+ "apply_spec_augment": false,
5
+ "architectures": [
6
+ "WhisperForConditionalGeneration"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "bos_token_id": 50257,
10
+ "classifier_proj_size": 256,
11
+ "d_model": 384,
12
+ "decoder_attention_heads": 6,
13
+ "decoder_ffn_dim": 1536,
14
+ "decoder_layerdrop": 0.0,
15
+ "decoder_layers": 4,
16
+ "decoder_start_token_id": 50258,
17
+ "dropout": 0.0,
18
+ "dtype": "float32",
19
+ "encoder_attention_heads": 6,
20
+ "encoder_ffn_dim": 1536,
21
+ "encoder_layerdrop": 0.0,
22
+ "encoder_layers": 4,
23
+ "eos_token_id": 50257,
24
+ "forced_decoder_ids": null,
25
+ "init_std": 0.02,
26
+ "is_encoder_decoder": true,
27
+ "mask_feature_length": 10,
28
+ "mask_feature_min_masks": 0,
29
+ "mask_feature_prob": 0.0,
30
+ "mask_time_length": 10,
31
+ "mask_time_min_masks": 2,
32
+ "mask_time_prob": 0.05,
33
+ "max_source_positions": 1500,
34
+ "max_target_positions": 448,
35
+ "median_filter_width": 7,
36
+ "model_type": "whisper",
37
+ "num_hidden_layers": 4,
38
+ "num_mel_bins": 80,
39
+ "pad_token_id": 50257,
40
+ "scale_embedding": false,
41
+ "tie_word_embeddings": true,
42
+ "transformers_version": "5.0.0",
43
+ "use_cache": false,
44
+ "use_weighted_layer_sum": false,
45
+ "vocab_size": 51865
46
+ }
generation_config.json ADDED
@@ -0,0 +1,241 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alignment_heads": [
3
+ [
4
+ 2,
5
+ 2
6
+ ],
7
+ [
8
+ 3,
9
+ 0
10
+ ],
11
+ [
12
+ 3,
13
+ 2
14
+ ],
15
+ [
16
+ 3,
17
+ 3
18
+ ],
19
+ [
20
+ 3,
21
+ 4
22
+ ],
23
+ [
24
+ 3,
25
+ 5
26
+ ]
27
+ ],
28
+ "begin_suppress_tokens": [
29
+ 220,
30
+ 50257
31
+ ],
32
+ "bos_token_id": 50257,
33
+ "decoder_start_token_id": 50258,
34
+ "eos_token_id": [
35
+ 50257
36
+ ],
37
+ "forced_decoder_ids": null,
38
+ "is_multilingual": true,
39
+ "lang_to_id": {
40
+ "<|af|>": 50327,
41
+ "<|am|>": 50334,
42
+ "<|ar|>": 50272,
43
+ "<|as|>": 50350,
44
+ "<|az|>": 50304,
45
+ "<|ba|>": 50355,
46
+ "<|be|>": 50330,
47
+ "<|bg|>": 50292,
48
+ "<|bn|>": 50302,
49
+ "<|bo|>": 50347,
50
+ "<|br|>": 50309,
51
+ "<|bs|>": 50315,
52
+ "<|ca|>": 50270,
53
+ "<|cs|>": 50283,
54
+ "<|cy|>": 50297,
55
+ "<|da|>": 50285,
56
+ "<|de|>": 50261,
57
+ "<|el|>": 50281,
58
+ "<|en|>": 50259,
59
+ "<|es|>": 50262,
60
+ "<|et|>": 50307,
61
+ "<|eu|>": 50310,
62
+ "<|fa|>": 50300,
63
+ "<|fi|>": 50277,
64
+ "<|fo|>": 50338,
65
+ "<|fr|>": 50265,
66
+ "<|gl|>": 50319,
67
+ "<|gu|>": 50333,
68
+ "<|haw|>": 50352,
69
+ "<|ha|>": 50354,
70
+ "<|he|>": 50279,
71
+ "<|hi|>": 50276,
72
+ "<|hr|>": 50291,
73
+ "<|ht|>": 50339,
74
+ "<|hu|>": 50286,
75
+ "<|hy|>": 50312,
76
+ "<|id|>": 50275,
77
+ "<|is|>": 50311,
78
+ "<|it|>": 50274,
79
+ "<|ja|>": 50266,
80
+ "<|jw|>": 50356,
81
+ "<|ka|>": 50329,
82
+ "<|kk|>": 50316,
83
+ "<|km|>": 50323,
84
+ "<|kn|>": 50306,
85
+ "<|ko|>": 50264,
86
+ "<|la|>": 50294,
87
+ "<|lb|>": 50345,
88
+ "<|ln|>": 50353,
89
+ "<|lo|>": 50336,
90
+ "<|lt|>": 50293,
91
+ "<|lv|>": 50301,
92
+ "<|mg|>": 50349,
93
+ "<|mi|>": 50295,
94
+ "<|mk|>": 50308,
95
+ "<|ml|>": 50296,
96
+ "<|mn|>": 50314,
97
+ "<|mr|>": 50320,
98
+ "<|ms|>": 50282,
99
+ "<|mt|>": 50343,
100
+ "<|my|>": 50346,
101
+ "<|ne|>": 50313,
102
+ "<|nl|>": 50271,
103
+ "<|nn|>": 50342,
104
+ "<|no|>": 50288,
105
+ "<|oc|>": 50328,
106
+ "<|pa|>": 50321,
107
+ "<|pl|>": 50269,
108
+ "<|ps|>": 50340,
109
+ "<|pt|>": 50267,
110
+ "<|ro|>": 50284,
111
+ "<|ru|>": 50263,
112
+ "<|sa|>": 50344,
113
+ "<|sd|>": 50332,
114
+ "<|si|>": 50322,
115
+ "<|sk|>": 50298,
116
+ "<|sl|>": 50305,
117
+ "<|sn|>": 50324,
118
+ "<|so|>": 50326,
119
+ "<|sq|>": 50317,
120
+ "<|sr|>": 50303,
121
+ "<|su|>": 50357,
122
+ "<|sv|>": 50273,
123
+ "<|sw|>": 50318,
124
+ "<|ta|>": 50287,
125
+ "<|te|>": 50299,
126
+ "<|tg|>": 50331,
127
+ "<|th|>": 50289,
128
+ "<|tk|>": 50341,
129
+ "<|tl|>": 50348,
130
+ "<|tr|>": 50268,
131
+ "<|tt|>": 50351,
132
+ "<|uk|>": 50280,
133
+ "<|ur|>": 50290,
134
+ "<|uz|>": 50337,
135
+ "<|vi|>": 50278,
136
+ "<|yi|>": 50335,
137
+ "<|yo|>": 50325,
138
+ "<|zh|>": 50260
139
+ },
140
+ "max_initial_timestamp_index": 50,
141
+ "max_length": 448,
142
+ "no_timestamps_token_id": 50363,
143
+ "pad_token_id": 50257,
144
+ "prev_sot_token_id": 50361,
145
+ "return_timestamps": false,
146
+ "suppress_tokens": [
147
+ 1,
148
+ 2,
149
+ 7,
150
+ 8,
151
+ 9,
152
+ 10,
153
+ 14,
154
+ 25,
155
+ 26,
156
+ 27,
157
+ 28,
158
+ 29,
159
+ 31,
160
+ 58,
161
+ 59,
162
+ 60,
163
+ 61,
164
+ 62,
165
+ 63,
166
+ 90,
167
+ 91,
168
+ 92,
169
+ 93,
170
+ 359,
171
+ 503,
172
+ 522,
173
+ 542,
174
+ 873,
175
+ 893,
176
+ 902,
177
+ 918,
178
+ 922,
179
+ 931,
180
+ 1350,
181
+ 1853,
182
+ 1982,
183
+ 2460,
184
+ 2627,
185
+ 3246,
186
+ 3253,
187
+ 3268,
188
+ 3536,
189
+ 3846,
190
+ 3961,
191
+ 4183,
192
+ 4667,
193
+ 6585,
194
+ 6647,
195
+ 7273,
196
+ 9061,
197
+ 9383,
198
+ 10428,
199
+ 10929,
200
+ 11938,
201
+ 12033,
202
+ 12331,
203
+ 12562,
204
+ 13793,
205
+ 14157,
206
+ 14635,
207
+ 15265,
208
+ 15618,
209
+ 16553,
210
+ 16604,
211
+ 18362,
212
+ 18956,
213
+ 20075,
214
+ 21675,
215
+ 22520,
216
+ 26130,
217
+ 26161,
218
+ 26435,
219
+ 28279,
220
+ 29464,
221
+ 31650,
222
+ 32302,
223
+ 32470,
224
+ 36865,
225
+ 42863,
226
+ 47425,
227
+ 49870,
228
+ 50254,
229
+ 50258,
230
+ 50358,
231
+ 50359,
232
+ 50360,
233
+ 50361,
234
+ 50362
235
+ ],
236
+ "task_to_id": {
237
+ "transcribe": 50359,
238
+ "translate": 50358
239
+ },
240
+ "transformers_version": "5.0.0"
241
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7024926f2124fcb40907f91584f20990731f7273387e65006e9d3e6953fc2b9
3
+ size 151061672
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a23514630188c39afda4d4f4cc24cb91b7964791b29289eade3ac7807e4999ef
3
+ size 302220811
processor_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "feature_extractor": {
3
+ "chunk_length": 30,
4
+ "dither": 0.0,
5
+ "feature_extractor_type": "WhisperFeatureExtractor",
6
+ "feature_size": 80,
7
+ "hop_length": 160,
8
+ "n_fft": 400,
9
+ "n_samples": 480000,
10
+ "nb_max_frames": 3000,
11
+ "padding_side": "right",
12
+ "padding_value": 0.0,
13
+ "return_attention_mask": false,
14
+ "sampling_rate": 16000
15
+ },
16
+ "processor_class": "WhisperProcessor"
17
+ }
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:039970b6882c953a4975fc4dfcb09d0a09918844f38de8373269a68c0dfa9305
3
+ size 14455
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a16c63c5cfdc9a7ee451fba1e6666f4cc178e3589a9ce9b2d4f69bdcbda86e79
3
+ size 1465
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|endoftext|>",
5
+ "clean_up_tokenization_spaces": true,
6
+ "eos_token": "<|endoftext|>",
7
+ "errors": "replace",
8
+ "extra_special_tokens": [
9
+ "<|endoftext|>",
10
+ "<|startoftranscript|>",
11
+ "<|en|>",
12
+ "<|zh|>",
13
+ "<|de|>",
14
+ "<|es|>",
15
+ "<|ru|>",
16
+ "<|ko|>",
17
+ "<|fr|>",
18
+ "<|ja|>",
19
+ "<|pt|>",
20
+ "<|tr|>",
21
+ "<|pl|>",
22
+ "<|ca|>",
23
+ "<|nl|>",
24
+ "<|ar|>",
25
+ "<|sv|>",
26
+ "<|it|>",
27
+ "<|id|>",
28
+ "<|hi|>",
29
+ "<|fi|>",
30
+ "<|vi|>",
31
+ "<|he|>",
32
+ "<|uk|>",
33
+ "<|el|>",
34
+ "<|ms|>",
35
+ "<|cs|>",
36
+ "<|ro|>",
37
+ "<|da|>",
38
+ "<|hu|>",
39
+ "<|ta|>",
40
+ "<|no|>",
41
+ "<|th|>",
42
+ "<|ur|>",
43
+ "<|hr|>",
44
+ "<|bg|>",
45
+ "<|lt|>",
46
+ "<|la|>",
47
+ "<|mi|>",
48
+ "<|ml|>",
49
+ "<|cy|>",
50
+ "<|sk|>",
51
+ "<|te|>",
52
+ "<|fa|>",
53
+ "<|lv|>",
54
+ "<|bn|>",
55
+ "<|sr|>",
56
+ "<|az|>",
57
+ "<|sl|>",
58
+ "<|kn|>",
59
+ "<|et|>",
60
+ "<|mk|>",
61
+ "<|br|>",
62
+ "<|eu|>",
63
+ "<|is|>",
64
+ "<|hy|>",
65
+ "<|ne|>",
66
+ "<|mn|>",
67
+ "<|bs|>",
68
+ "<|kk|>",
69
+ "<|sq|>",
70
+ "<|sw|>",
71
+ "<|gl|>",
72
+ "<|mr|>",
73
+ "<|pa|>",
74
+ "<|si|>",
75
+ "<|km|>",
76
+ "<|sn|>",
77
+ "<|yo|>",
78
+ "<|so|>",
79
+ "<|af|>",
80
+ "<|oc|>",
81
+ "<|ka|>",
82
+ "<|be|>",
83
+ "<|tg|>",
84
+ "<|sd|>",
85
+ "<|gu|>",
86
+ "<|am|>",
87
+ "<|yi|>",
88
+ "<|lo|>",
89
+ "<|uz|>",
90
+ "<|fo|>",
91
+ "<|ht|>",
92
+ "<|ps|>",
93
+ "<|tk|>",
94
+ "<|nn|>",
95
+ "<|mt|>",
96
+ "<|sa|>",
97
+ "<|lb|>",
98
+ "<|my|>",
99
+ "<|bo|>",
100
+ "<|tl|>",
101
+ "<|mg|>",
102
+ "<|as|>",
103
+ "<|tt|>",
104
+ "<|haw|>",
105
+ "<|ln|>",
106
+ "<|ha|>",
107
+ "<|ba|>",
108
+ "<|jw|>",
109
+ "<|su|>",
110
+ "<|translate|>",
111
+ "<|transcribe|>",
112
+ "<|startoflm|>",
113
+ "<|startofprev|>",
114
+ "<|nocaptions|>",
115
+ "<|notimestamps|>"
116
+ ],
117
+ "is_local": false,
118
+ "language": "french",
119
+ "model_max_length": 1024,
120
+ "pad_token": "<|endoftext|>",
121
+ "predict_timestamps": false,
122
+ "processor_class": "WhisperProcessor",
123
+ "return_attention_mask": false,
124
+ "task": "translate",
125
+ "tokenizer_class": "WhisperTokenizer",
126
+ "unk_token": "<|endoftext|>"
127
+ }
trainer_state.json ADDED
@@ -0,0 +1,254 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 500,
3
+ "best_metric": 2.7007479667663574,
4
+ "best_model_checkpoint": "Mazzy-whisper-tiny\\checkpoint-500",
5
+ "epoch": 0.625,
6
+ "eval_steps": 50,
7
+ "global_step": 500,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.03125,
14
+ "grad_norm": 75.82250213623047,
15
+ "learning_rate": 4.800000000000001e-06,
16
+ "loss": 4.257205200195313,
17
+ "step": 25
18
+ },
19
+ {
20
+ "epoch": 0.0625,
21
+ "grad_norm": 91.59275817871094,
22
+ "learning_rate": 9.800000000000001e-06,
23
+ "loss": 3.810027770996094,
24
+ "step": 50
25
+ },
26
+ {
27
+ "epoch": 0.0625,
28
+ "eval_loss": 3.387531042098999,
29
+ "eval_runtime": 56.07,
30
+ "eval_samples_per_second": 1.783,
31
+ "eval_steps_per_second": 1.783,
32
+ "step": 50
33
+ },
34
+ {
35
+ "epoch": 0.09375,
36
+ "grad_norm": 109.6454086303711,
37
+ "learning_rate": 9.466666666666667e-06,
38
+ "loss": 3.426136169433594,
39
+ "step": 75
40
+ },
41
+ {
42
+ "epoch": 0.125,
43
+ "grad_norm": 77.40877532958984,
44
+ "learning_rate": 8.91111111111111e-06,
45
+ "loss": 3.1471734619140626,
46
+ "step": 100
47
+ },
48
+ {
49
+ "epoch": 0.125,
50
+ "eval_loss": 3.086820602416992,
51
+ "eval_runtime": 55.5379,
52
+ "eval_samples_per_second": 1.801,
53
+ "eval_steps_per_second": 1.801,
54
+ "step": 100
55
+ },
56
+ {
57
+ "epoch": 0.15625,
58
+ "grad_norm": 65.8832015991211,
59
+ "learning_rate": 8.355555555555556e-06,
60
+ "loss": 2.979889221191406,
61
+ "step": 125
62
+ },
63
+ {
64
+ "epoch": 0.1875,
65
+ "grad_norm": 97.50338745117188,
66
+ "learning_rate": 7.800000000000002e-06,
67
+ "loss": 3.0117352294921873,
68
+ "step": 150
69
+ },
70
+ {
71
+ "epoch": 0.1875,
72
+ "eval_loss": 2.8795320987701416,
73
+ "eval_runtime": 56.4467,
74
+ "eval_samples_per_second": 1.772,
75
+ "eval_steps_per_second": 1.772,
76
+ "step": 150
77
+ },
78
+ {
79
+ "epoch": 0.21875,
80
+ "grad_norm": 63.26200866699219,
81
+ "learning_rate": 7.244444444444445e-06,
82
+ "loss": 2.979839172363281,
83
+ "step": 175
84
+ },
85
+ {
86
+ "epoch": 0.25,
87
+ "grad_norm": 56.74595642089844,
88
+ "learning_rate": 6.688888888888889e-06,
89
+ "loss": 2.9879776000976563,
90
+ "step": 200
91
+ },
92
+ {
93
+ "epoch": 0.25,
94
+ "eval_loss": 2.8465237617492676,
95
+ "eval_runtime": 76.306,
96
+ "eval_samples_per_second": 1.311,
97
+ "eval_steps_per_second": 1.311,
98
+ "step": 200
99
+ },
100
+ {
101
+ "epoch": 0.28125,
102
+ "grad_norm": 96.1812515258789,
103
+ "learning_rate": 6.133333333333334e-06,
104
+ "loss": 2.7666366577148436,
105
+ "step": 225
106
+ },
107
+ {
108
+ "epoch": 0.3125,
109
+ "grad_norm": 71.51004791259766,
110
+ "learning_rate": 5.577777777777778e-06,
111
+ "loss": 2.8252435302734376,
112
+ "step": 250
113
+ },
114
+ {
115
+ "epoch": 0.3125,
116
+ "eval_loss": 2.7943179607391357,
117
+ "eval_runtime": 1293.0599,
118
+ "eval_samples_per_second": 0.077,
119
+ "eval_steps_per_second": 0.077,
120
+ "step": 250
121
+ },
122
+ {
123
+ "epoch": 0.34375,
124
+ "grad_norm": 98.3506088256836,
125
+ "learning_rate": 5.022222222222223e-06,
126
+ "loss": 2.8509982299804686,
127
+ "step": 275
128
+ },
129
+ {
130
+ "epoch": 0.375,
131
+ "grad_norm": 76.27555847167969,
132
+ "learning_rate": 4.4666666666666665e-06,
133
+ "loss": 2.8936077880859377,
134
+ "step": 300
135
+ },
136
+ {
137
+ "epoch": 0.375,
138
+ "eval_loss": 2.7515389919281006,
139
+ "eval_runtime": 247.7616,
140
+ "eval_samples_per_second": 0.404,
141
+ "eval_steps_per_second": 0.404,
142
+ "step": 300
143
+ },
144
+ {
145
+ "epoch": 0.40625,
146
+ "grad_norm": 115.49108123779297,
147
+ "learning_rate": 3.911111111111112e-06,
148
+ "loss": 2.7140411376953124,
149
+ "step": 325
150
+ },
151
+ {
152
+ "epoch": 0.4375,
153
+ "grad_norm": 62.67667007446289,
154
+ "learning_rate": 3.3555555555555557e-06,
155
+ "loss": 2.800578308105469,
156
+ "step": 350
157
+ },
158
+ {
159
+ "epoch": 0.4375,
160
+ "eval_loss": 2.7254576683044434,
161
+ "eval_runtime": 52.2594,
162
+ "eval_samples_per_second": 1.914,
163
+ "eval_steps_per_second": 1.914,
164
+ "step": 350
165
+ },
166
+ {
167
+ "epoch": 0.46875,
168
+ "grad_norm": 53.711524963378906,
169
+ "learning_rate": 2.8000000000000003e-06,
170
+ "loss": 2.7111419677734374,
171
+ "step": 375
172
+ },
173
+ {
174
+ "epoch": 0.5,
175
+ "grad_norm": 55.54822540283203,
176
+ "learning_rate": 2.2444444444444445e-06,
177
+ "loss": 2.8701107788085936,
178
+ "step": 400
179
+ },
180
+ {
181
+ "epoch": 0.5,
182
+ "eval_loss": 2.7176525592803955,
183
+ "eval_runtime": 55.6682,
184
+ "eval_samples_per_second": 1.796,
185
+ "eval_steps_per_second": 1.796,
186
+ "step": 400
187
+ },
188
+ {
189
+ "epoch": 0.53125,
190
+ "grad_norm": 106.42744445800781,
191
+ "learning_rate": 1.688888888888889e-06,
192
+ "loss": 2.784833984375,
193
+ "step": 425
194
+ },
195
+ {
196
+ "epoch": 0.5625,
197
+ "grad_norm": 55.58470153808594,
198
+ "learning_rate": 1.1333333333333334e-06,
199
+ "loss": 2.8060418701171876,
200
+ "step": 450
201
+ },
202
+ {
203
+ "epoch": 0.5625,
204
+ "eval_loss": 2.7064270973205566,
205
+ "eval_runtime": 76.5646,
206
+ "eval_samples_per_second": 1.306,
207
+ "eval_steps_per_second": 1.306,
208
+ "step": 450
209
+ },
210
+ {
211
+ "epoch": 0.59375,
212
+ "grad_norm": 49.802032470703125,
213
+ "learning_rate": 5.777777777777778e-07,
214
+ "loss": 2.7749835205078126,
215
+ "step": 475
216
+ },
217
+ {
218
+ "epoch": 0.625,
219
+ "grad_norm": 69.92212677001953,
220
+ "learning_rate": 2.2222222222222224e-08,
221
+ "loss": 2.775958251953125,
222
+ "step": 500
223
+ },
224
+ {
225
+ "epoch": 0.625,
226
+ "eval_loss": 2.7007479667663574,
227
+ "eval_runtime": 53.3461,
228
+ "eval_samples_per_second": 1.875,
229
+ "eval_steps_per_second": 1.875,
230
+ "step": 500
231
+ }
232
+ ],
233
+ "logging_steps": 25,
234
+ "max_steps": 500,
235
+ "num_input_tokens_seen": 0,
236
+ "num_train_epochs": 1,
237
+ "save_steps": 50,
238
+ "stateful_callbacks": {
239
+ "TrainerControl": {
240
+ "args": {
241
+ "should_epoch_stop": false,
242
+ "should_evaluate": false,
243
+ "should_log": false,
244
+ "should_save": true,
245
+ "should_training_stop": true
246
+ },
247
+ "attributes": {}
248
+ }
249
+ },
250
+ "total_flos": 1.230944256e+16,
251
+ "train_batch_size": 1,
252
+ "trial_name": null,
253
+ "trial_params": null
254
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4827c7e5dee97d95625fff6cd366043a4c3dfbe65cbe0a8812baa00c4a50c05c
3
+ size 5393