File size: 6,464 Bytes
4dc7c40
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
35
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
{
  "best_metric": 0.7326732673267327,
  "best_model_checkpoint": "distilhubert-finetuned-not-a-word2/run-8/checkpoint-24",
  "epoch": 6.0,
  "eval_steps": 500,
  "global_step": 144,
  "is_hyper_param_search": true,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.21,
      "grad_norm": 1.6496813297271729,
      "learning_rate": 2.9441483262927863e-06,
      "loss": 0.6992,
      "step": 5
    },
    {
      "epoch": 0.42,
      "grad_norm": 0.620004415512085,
      "learning_rate": 5.888296652585573e-06,
      "loss": 0.6939,
      "step": 10
    },
    {
      "epoch": 0.62,
      "grad_norm": 0.5726878046989441,
      "learning_rate": 8.832444978878358e-06,
      "loss": 0.6835,
      "step": 15
    },
    {
      "epoch": 0.83,
      "grad_norm": 1.4239176511764526,
      "learning_rate": 1.1776593305171145e-05,
      "loss": 0.6673,
      "step": 20
    },
    {
      "epoch": 1.0,
      "eval_f1": 0.7326732673267327,
      "eval_loss": 0.6805419921875,
      "eval_runtime": 1.3716,
      "eval_samples_per_second": 46.661,
      "eval_steps_per_second": 5.833,
      "step": 24
    },
    {
      "epoch": 1.04,
      "grad_norm": 0.9195191264152527,
      "learning_rate": 1.406648644784331e-05,
      "loss": 0.6673,
      "step": 25
    },
    {
      "epoch": 1.25,
      "grad_norm": 1.8524231910705566,
      "learning_rate": 1.3739358856033002e-05,
      "loss": 0.6155,
      "step": 30
    },
    {
      "epoch": 1.46,
      "grad_norm": 1.8213531970977783,
      "learning_rate": 1.3412231264222692e-05,
      "loss": 0.5895,
      "step": 35
    },
    {
      "epoch": 1.67,
      "grad_norm": 0.4818130433559418,
      "learning_rate": 1.3085103672412383e-05,
      "loss": 0.6468,
      "step": 40
    },
    {
      "epoch": 1.88,
      "grad_norm": 0.6597484946250916,
      "learning_rate": 1.2757976080602073e-05,
      "loss": 0.6173,
      "step": 45
    },
    {
      "epoch": 2.0,
      "eval_f1": 0.7326732673267327,
      "eval_loss": 0.6970634460449219,
      "eval_runtime": 1.3632,
      "eval_samples_per_second": 46.949,
      "eval_steps_per_second": 5.869,
      "step": 48
    },
    {
      "epoch": 2.08,
      "grad_norm": 0.48903289437294006,
      "learning_rate": 1.2430848488791764e-05,
      "loss": 0.6302,
      "step": 50
    },
    {
      "epoch": 2.29,
      "grad_norm": 0.6064260601997375,
      "learning_rate": 1.2103720896981454e-05,
      "loss": 0.5867,
      "step": 55
    },
    {
      "epoch": 2.5,
      "grad_norm": 0.6802453398704529,
      "learning_rate": 1.1776593305171145e-05,
      "loss": 0.6321,
      "step": 60
    },
    {
      "epoch": 2.71,
      "grad_norm": 1.2592875957489014,
      "learning_rate": 1.1449465713360835e-05,
      "loss": 0.6223,
      "step": 65
    },
    {
      "epoch": 2.92,
      "grad_norm": 1.1591824293136597,
      "learning_rate": 1.1122338121550526e-05,
      "loss": 0.4922,
      "step": 70
    },
    {
      "epoch": 3.0,
      "eval_f1": 0.7326732673267327,
      "eval_loss": 0.7079887390136719,
      "eval_runtime": 1.3669,
      "eval_samples_per_second": 46.821,
      "eval_steps_per_second": 5.853,
      "step": 72
    },
    {
      "epoch": 3.12,
      "grad_norm": 0.5332023501396179,
      "learning_rate": 1.0795210529740214e-05,
      "loss": 0.5989,
      "step": 75
    },
    {
      "epoch": 3.33,
      "grad_norm": 0.5555600523948669,
      "learning_rate": 1.0468082937929906e-05,
      "loss": 0.6104,
      "step": 80
    },
    {
      "epoch": 3.54,
      "grad_norm": 1.2928024530410767,
      "learning_rate": 1.0140955346119596e-05,
      "loss": 0.4936,
      "step": 85
    },
    {
      "epoch": 3.75,
      "grad_norm": 1.1424989700317383,
      "learning_rate": 9.813827754309287e-06,
      "loss": 0.6191,
      "step": 90
    },
    {
      "epoch": 3.96,
      "grad_norm": 1.119732141494751,
      "learning_rate": 9.486700162498977e-06,
      "loss": 0.6004,
      "step": 95
    },
    {
      "epoch": 4.0,
      "eval_f1": 0.7326732673267327,
      "eval_loss": 0.7053489685058594,
      "eval_runtime": 1.3556,
      "eval_samples_per_second": 47.211,
      "eval_steps_per_second": 5.901,
      "step": 96
    },
    {
      "epoch": 4.17,
      "grad_norm": 0.8135461211204529,
      "learning_rate": 9.159572570688668e-06,
      "loss": 0.5154,
      "step": 100
    },
    {
      "epoch": 4.38,
      "grad_norm": 1.8034342527389526,
      "learning_rate": 8.832444978878358e-06,
      "loss": 0.6067,
      "step": 105
    },
    {
      "epoch": 4.58,
      "grad_norm": 0.8029685020446777,
      "learning_rate": 8.505317387068049e-06,
      "loss": 0.5499,
      "step": 110
    },
    {
      "epoch": 4.79,
      "grad_norm": 1.019626259803772,
      "learning_rate": 8.178189795257739e-06,
      "loss": 0.5542,
      "step": 115
    },
    {
      "epoch": 5.0,
      "grad_norm": 1.861674427986145,
      "learning_rate": 7.85106220344743e-06,
      "loss": 0.5545,
      "step": 120
    },
    {
      "epoch": 5.0,
      "eval_f1": 0.7326732673267327,
      "eval_loss": 0.6832504272460938,
      "eval_runtime": 1.3811,
      "eval_samples_per_second": 46.341,
      "eval_steps_per_second": 5.793,
      "step": 120
    },
    {
      "epoch": 5.21,
      "grad_norm": 1.5949212312698364,
      "learning_rate": 7.523934611637121e-06,
      "loss": 0.4806,
      "step": 125
    },
    {
      "epoch": 5.42,
      "grad_norm": 3.002861738204956,
      "learning_rate": 7.196807019826811e-06,
      "loss": 0.5832,
      "step": 130
    },
    {
      "epoch": 5.62,
      "grad_norm": 1.4606820344924927,
      "learning_rate": 6.9351049463785626e-06,
      "loss": 0.5481,
      "step": 135
    },
    {
      "epoch": 5.83,
      "grad_norm": 1.6088628768920898,
      "learning_rate": 6.607977354568253e-06,
      "loss": 0.5333,
      "step": 140
    },
    {
      "epoch": 6.0,
      "eval_f1": 0.7326732673267327,
      "eval_loss": 0.6599597930908203,
      "eval_runtime": 1.3747,
      "eval_samples_per_second": 46.555,
      "eval_steps_per_second": 5.819,
      "step": 144
    }
  ],
  "logging_steps": 5,
  "max_steps": 240,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 10,
  "save_steps": 500,
  "total_flos": 4911381340990080.0,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": {
    "learning_rate": 1.4131911966205373e-05,
    "per_device_train_batch_size": 8
  }
}