kevinoli commited on
Commit
1db3703
1 Parent(s): 83280b4

Training in progress, step 9000, checkpoint

Browse files
checkpoint-9000/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a712e3721561d8db5c426b8e1f1d86dc1944176cde2e6df4cf31ead39383eb24
3
  size 1711848436
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c7006edb64df6aeea55de1d9383a1d5ba7df07e0289eae0b67e50d27f10834f
3
  size 1711848436
checkpoint-9000/optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cd1a5bbd4d8969ea06b43aca88d1b810210f39d04e6c6ab4145bb0538312cff9
3
  size 3424043887
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77ab61faab3cb01fd64b5f30980bee98d3009c9a4ce9100e935bd2c31f93268f
3
  size 3424043887
checkpoint-9000/trainer_state.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "best_metric": 0.9543817043304443,
3
  "best_model_checkpoint": "./output/clip-finetuned-csu-p14-336-e4l58-l/checkpoint-9000",
4
  "epoch": 1.658069270449521,
5
  "eval_steps": 500,
@@ -10,272 +10,272 @@
10
  "log_history": [
11
  {
12
  "epoch": 0.09211495946941783,
13
- "grad_norm": 20.28152847290039,
14
  "learning_rate": 4.8848563006632274e-08,
15
  "loss": 0.3758,
16
  "step": 500
17
  },
18
  {
19
  "epoch": 0.09211495946941783,
20
- "eval_loss": 1.418367862701416,
21
- "eval_runtime": 73.4439,
22
- "eval_samples_per_second": 16.434,
23
- "eval_steps_per_second": 2.056,
24
  "step": 500
25
  },
26
  {
27
  "epoch": 0.18422991893883567,
28
- "grad_norm": 47.29538345336914,
29
  "learning_rate": 4.7697126013264556e-08,
30
- "loss": 0.4104,
31
  "step": 1000
32
  },
33
  {
34
  "epoch": 0.18422991893883567,
35
- "eval_loss": 1.3491536378860474,
36
- "eval_runtime": 74.3404,
37
- "eval_samples_per_second": 16.236,
38
- "eval_steps_per_second": 2.031,
39
  "step": 1000
40
  },
41
  {
42
  "epoch": 0.2763448784082535,
43
- "grad_norm": 0.5450155138969421,
44
  "learning_rate": 4.6545689019896826e-08,
45
- "loss": 0.4325,
46
  "step": 1500
47
  },
48
  {
49
  "epoch": 0.2763448784082535,
50
- "eval_loss": 1.2877168655395508,
51
- "eval_runtime": 75.3303,
52
- "eval_samples_per_second": 16.023,
53
- "eval_steps_per_second": 2.005,
54
  "step": 1500
55
  },
56
  {
57
  "epoch": 0.36845983787767134,
58
- "grad_norm": 0.11449883133172989,
59
  "learning_rate": 4.539425202652911e-08,
60
- "loss": 0.3423,
61
  "step": 2000
62
  },
63
  {
64
  "epoch": 0.36845983787767134,
65
- "eval_loss": 1.2384389638900757,
66
- "eval_runtime": 76.113,
67
- "eval_samples_per_second": 15.858,
68
- "eval_steps_per_second": 1.984,
69
  "step": 2000
70
  },
71
  {
72
  "epoch": 0.46057479734708917,
73
- "grad_norm": 389.7380065917969,
74
  "learning_rate": 4.4242815033161385e-08,
75
  "loss": 0.3645,
76
  "step": 2500
77
  },
78
  {
79
  "epoch": 0.46057479734708917,
80
- "eval_loss": 1.1896915435791016,
81
- "eval_runtime": 76.23,
82
- "eval_samples_per_second": 15.834,
83
- "eval_steps_per_second": 1.981,
84
  "step": 2500
85
  },
86
  {
87
  "epoch": 0.552689756816507,
88
- "grad_norm": 340.1103820800781,
89
  "learning_rate": 4.309137803979366e-08,
90
- "loss": 0.3171,
91
  "step": 3000
92
  },
93
  {
94
  "epoch": 0.552689756816507,
95
- "eval_loss": 1.149291753768921,
96
- "eval_runtime": 76.1662,
97
- "eval_samples_per_second": 15.847,
98
- "eval_steps_per_second": 1.983,
99
  "step": 3000
100
  },
101
  {
102
  "epoch": 0.6448047162859248,
103
- "grad_norm": 196.7362518310547,
104
  "learning_rate": 4.193994104642594e-08,
105
- "loss": 0.275,
106
  "step": 3500
107
  },
108
  {
109
  "epoch": 0.6448047162859248,
110
- "eval_loss": 1.1162768602371216,
111
- "eval_runtime": 74.6569,
112
- "eval_samples_per_second": 16.167,
113
- "eval_steps_per_second": 2.023,
114
  "step": 3500
115
  },
116
  {
117
  "epoch": 0.7369196757553427,
118
- "grad_norm": 16.604211807250977,
119
  "learning_rate": 4.078850405305821e-08,
120
- "loss": 0.2907,
121
  "step": 4000
122
  },
123
  {
124
  "epoch": 0.7369196757553427,
125
- "eval_loss": 1.091786503791809,
126
- "eval_runtime": 76.3333,
127
- "eval_samples_per_second": 15.812,
128
- "eval_steps_per_second": 1.978,
129
  "step": 4000
130
  },
131
  {
132
  "epoch": 0.8290346352247605,
133
- "grad_norm": 527.0987548828125,
134
  "learning_rate": 3.9637067059690496e-08,
135
- "loss": 0.2581,
136
  "step": 4500
137
  },
138
  {
139
  "epoch": 0.8290346352247605,
140
- "eval_loss": 1.0729899406433105,
141
- "eval_runtime": 74.4912,
142
- "eval_samples_per_second": 16.203,
143
- "eval_steps_per_second": 2.027,
144
  "step": 4500
145
  },
146
  {
147
  "epoch": 0.9211495946941783,
148
- "grad_norm": 1.148181676864624,
149
  "learning_rate": 3.848563006632277e-08,
150
- "loss": 0.268,
151
  "step": 5000
152
  },
153
  {
154
  "epoch": 0.9211495946941783,
155
- "eval_loss": 1.052911400794983,
156
- "eval_runtime": 74.9586,
157
- "eval_samples_per_second": 16.102,
158
- "eval_steps_per_second": 2.014,
159
  "step": 5000
160
  },
161
  {
162
  "epoch": 1.013264554163596,
163
- "grad_norm": 0.1680648922920227,
164
  "learning_rate": 3.733419307295505e-08,
165
- "loss": 0.2467,
166
  "step": 5500
167
  },
168
  {
169
  "epoch": 1.013264554163596,
170
- "eval_loss": 1.036412239074707,
171
- "eval_runtime": 74.707,
172
- "eval_samples_per_second": 16.156,
173
- "eval_steps_per_second": 2.021,
174
  "step": 5500
175
  },
176
  {
177
  "epoch": 1.105379513633014,
178
- "grad_norm": 269.06585693359375,
179
  "learning_rate": 3.6182756079587324e-08,
180
- "loss": 0.2594,
181
  "step": 6000
182
  },
183
  {
184
  "epoch": 1.105379513633014,
185
- "eval_loss": 1.0190415382385254,
186
- "eval_runtime": 74.5657,
187
- "eval_samples_per_second": 16.187,
188
- "eval_steps_per_second": 2.025,
189
  "step": 6000
190
  },
191
  {
192
  "epoch": 1.1974944731024317,
193
- "grad_norm": 457.15850830078125,
194
  "learning_rate": 3.50313190862196e-08,
195
  "loss": 0.2153,
196
  "step": 6500
197
  },
198
  {
199
  "epoch": 1.1974944731024317,
200
- "eval_loss": 1.0054280757904053,
201
- "eval_runtime": 74.5269,
202
- "eval_samples_per_second": 16.195,
203
- "eval_steps_per_second": 2.026,
204
  "step": 6500
205
  },
206
  {
207
  "epoch": 1.2896094325718497,
208
- "grad_norm": 0.5831326842308044,
209
  "learning_rate": 3.387988209285188e-08,
210
- "loss": 0.1792,
211
  "step": 7000
212
  },
213
  {
214
  "epoch": 1.2896094325718497,
215
- "eval_loss": 0.9938780665397644,
216
- "eval_runtime": 74.5283,
217
- "eval_samples_per_second": 16.195,
218
- "eval_steps_per_second": 2.026,
219
  "step": 7000
220
  },
221
  {
222
  "epoch": 1.3817243920412676,
223
- "grad_norm": 54.12948226928711,
224
  "learning_rate": 3.272844509948415e-08,
225
- "loss": 0.208,
226
  "step": 7500
227
  },
228
  {
229
  "epoch": 1.3817243920412676,
230
- "eval_loss": 0.9816781282424927,
231
- "eval_runtime": 76.244,
232
- "eval_samples_per_second": 15.831,
233
- "eval_steps_per_second": 1.98,
234
  "step": 7500
235
  },
236
  {
237
  "epoch": 1.4738393515106853,
238
- "grad_norm": 0.09778845310211182,
239
  "learning_rate": 3.1577008106116435e-08,
240
- "loss": 0.1955,
241
  "step": 8000
242
  },
243
  {
244
  "epoch": 1.4738393515106853,
245
- "eval_loss": 0.9722194671630859,
246
- "eval_runtime": 74.4255,
247
- "eval_samples_per_second": 16.218,
248
- "eval_steps_per_second": 2.029,
249
  "step": 8000
250
  },
251
  {
252
  "epoch": 1.565954310980103,
253
- "grad_norm": 43.05193328857422,
254
  "learning_rate": 3.042557111274871e-08,
255
- "loss": 0.1943,
256
  "step": 8500
257
  },
258
  {
259
  "epoch": 1.565954310980103,
260
- "eval_loss": 0.962623119354248,
261
- "eval_runtime": 74.5615,
262
- "eval_samples_per_second": 16.188,
263
- "eval_steps_per_second": 2.025,
264
  "step": 8500
265
  },
266
  {
267
  "epoch": 1.658069270449521,
268
- "grad_norm": 0.050531528890132904,
269
  "learning_rate": 2.927413411938099e-08,
270
- "loss": 0.2421,
271
  "step": 9000
272
  },
273
  {
274
  "epoch": 1.658069270449521,
275
- "eval_loss": 0.9543817043304443,
276
- "eval_runtime": 76.2957,
277
- "eval_samples_per_second": 15.82,
278
- "eval_steps_per_second": 1.979,
279
  "step": 9000
280
  }
281
  ],
 
1
  {
2
+ "best_metric": 0.9536784291267395,
3
  "best_model_checkpoint": "./output/clip-finetuned-csu-p14-336-e4l58-l/checkpoint-9000",
4
  "epoch": 1.658069270449521,
5
  "eval_steps": 500,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.09211495946941783,
13
+ "grad_norm": 20.04043960571289,
14
  "learning_rate": 4.8848563006632274e-08,
15
  "loss": 0.3758,
16
  "step": 500
17
  },
18
  {
19
  "epoch": 0.09211495946941783,
20
+ "eval_loss": 1.4184927940368652,
21
+ "eval_runtime": 74.3562,
22
+ "eval_samples_per_second": 16.233,
23
+ "eval_steps_per_second": 2.031,
24
  "step": 500
25
  },
26
  {
27
  "epoch": 0.18422991893883567,
28
+ "grad_norm": 74.25025939941406,
29
  "learning_rate": 4.7697126013264556e-08,
30
+ "loss": 0.4103,
31
  "step": 1000
32
  },
33
  {
34
  "epoch": 0.18422991893883567,
35
+ "eval_loss": 1.3500770330429077,
36
+ "eval_runtime": 75.7013,
37
+ "eval_samples_per_second": 15.944,
38
+ "eval_steps_per_second": 1.995,
39
  "step": 1000
40
  },
41
  {
42
  "epoch": 0.2763448784082535,
43
+ "grad_norm": 0.5102410912513733,
44
  "learning_rate": 4.6545689019896826e-08,
45
+ "loss": 0.433,
46
  "step": 1500
47
  },
48
  {
49
  "epoch": 0.2763448784082535,
50
+ "eval_loss": 1.2885024547576904,
51
+ "eval_runtime": 75.197,
52
+ "eval_samples_per_second": 16.051,
53
+ "eval_steps_per_second": 2.008,
54
  "step": 1500
55
  },
56
  {
57
  "epoch": 0.36845983787767134,
58
+ "grad_norm": 0.1667678952217102,
59
  "learning_rate": 4.539425202652911e-08,
60
+ "loss": 0.3424,
61
  "step": 2000
62
  },
63
  {
64
  "epoch": 0.36845983787767134,
65
+ "eval_loss": 1.239119052886963,
66
+ "eval_runtime": 75.387,
67
+ "eval_samples_per_second": 16.011,
68
+ "eval_steps_per_second": 2.003,
69
  "step": 2000
70
  },
71
  {
72
  "epoch": 0.46057479734708917,
73
+ "grad_norm": 343.8175048828125,
74
  "learning_rate": 4.4242815033161385e-08,
75
  "loss": 0.3645,
76
  "step": 2500
77
  },
78
  {
79
  "epoch": 0.46057479734708917,
80
+ "eval_loss": 1.1902339458465576,
81
+ "eval_runtime": 75.9899,
82
+ "eval_samples_per_second": 15.884,
83
+ "eval_steps_per_second": 1.987,
84
  "step": 2500
85
  },
86
  {
87
  "epoch": 0.552689756816507,
88
+ "grad_norm": 305.5626525878906,
89
  "learning_rate": 4.309137803979366e-08,
90
+ "loss": 0.3172,
91
  "step": 3000
92
  },
93
  {
94
  "epoch": 0.552689756816507,
95
+ "eval_loss": 1.1506118774414062,
96
+ "eval_runtime": 74.8655,
97
+ "eval_samples_per_second": 16.122,
98
+ "eval_steps_per_second": 2.017,
99
  "step": 3000
100
  },
101
  {
102
  "epoch": 0.6448047162859248,
103
+ "grad_norm": 65.59048461914062,
104
  "learning_rate": 4.193994104642594e-08,
105
+ "loss": 0.2751,
106
  "step": 3500
107
  },
108
  {
109
  "epoch": 0.6448047162859248,
110
+ "eval_loss": 1.1169357299804688,
111
+ "eval_runtime": 76.2936,
112
+ "eval_samples_per_second": 15.82,
113
+ "eval_steps_per_second": 1.979,
114
  "step": 3500
115
  },
116
  {
117
  "epoch": 0.7369196757553427,
118
+ "grad_norm": 18.106201171875,
119
  "learning_rate": 4.078850405305821e-08,
120
+ "loss": 0.2919,
121
  "step": 4000
122
  },
123
  {
124
  "epoch": 0.7369196757553427,
125
+ "eval_loss": 1.0920671224594116,
126
+ "eval_runtime": 74.9101,
127
+ "eval_samples_per_second": 16.113,
128
+ "eval_steps_per_second": 2.016,
129
  "step": 4000
130
  },
131
  {
132
  "epoch": 0.8290346352247605,
133
+ "grad_norm": 506.40570068359375,
134
  "learning_rate": 3.9637067059690496e-08,
135
+ "loss": 0.2583,
136
  "step": 4500
137
  },
138
  {
139
  "epoch": 0.8290346352247605,
140
+ "eval_loss": 1.0721209049224854,
141
+ "eval_runtime": 75.8053,
142
+ "eval_samples_per_second": 15.922,
143
+ "eval_steps_per_second": 1.992,
144
  "step": 4500
145
  },
146
  {
147
  "epoch": 0.9211495946941783,
148
+ "grad_norm": 1.1508910655975342,
149
  "learning_rate": 3.848563006632277e-08,
150
+ "loss": 0.2679,
151
  "step": 5000
152
  },
153
  {
154
  "epoch": 0.9211495946941783,
155
+ "eval_loss": 1.0519349575042725,
156
+ "eval_runtime": 75.226,
157
+ "eval_samples_per_second": 16.045,
158
+ "eval_steps_per_second": 2.007,
159
  "step": 5000
160
  },
161
  {
162
  "epoch": 1.013264554163596,
163
+ "grad_norm": 0.1393290013074875,
164
  "learning_rate": 3.733419307295505e-08,
165
+ "loss": 0.2472,
166
  "step": 5500
167
  },
168
  {
169
  "epoch": 1.013264554163596,
170
+ "eval_loss": 1.0355629920959473,
171
+ "eval_runtime": 74.6535,
172
+ "eval_samples_per_second": 16.168,
173
+ "eval_steps_per_second": 2.023,
174
  "step": 5500
175
  },
176
  {
177
  "epoch": 1.105379513633014,
178
+ "grad_norm": 200.34664916992188,
179
  "learning_rate": 3.6182756079587324e-08,
180
+ "loss": 0.26,
181
  "step": 6000
182
  },
183
  {
184
  "epoch": 1.105379513633014,
185
+ "eval_loss": 1.0176944732666016,
186
+ "eval_runtime": 74.4374,
187
+ "eval_samples_per_second": 16.215,
188
+ "eval_steps_per_second": 2.029,
189
  "step": 6000
190
  },
191
  {
192
  "epoch": 1.1974944731024317,
193
+ "grad_norm": 431.1804504394531,
194
  "learning_rate": 3.50313190862196e-08,
195
  "loss": 0.2153,
196
  "step": 6500
197
  },
198
  {
199
  "epoch": 1.1974944731024317,
200
+ "eval_loss": 1.0044586658477783,
201
+ "eval_runtime": 74.3247,
202
+ "eval_samples_per_second": 16.24,
203
+ "eval_steps_per_second": 2.032,
204
  "step": 6500
205
  },
206
  {
207
  "epoch": 1.2896094325718497,
208
+ "grad_norm": 0.6204941272735596,
209
  "learning_rate": 3.387988209285188e-08,
210
+ "loss": 0.1791,
211
  "step": 7000
212
  },
213
  {
214
  "epoch": 1.2896094325718497,
215
+ "eval_loss": 0.9926707148551941,
216
+ "eval_runtime": 75.7317,
217
+ "eval_samples_per_second": 15.938,
218
+ "eval_steps_per_second": 1.994,
219
  "step": 7000
220
  },
221
  {
222
  "epoch": 1.3817243920412676,
223
+ "grad_norm": 52.27273178100586,
224
  "learning_rate": 3.272844509948415e-08,
225
+ "loss": 0.2082,
226
  "step": 7500
227
  },
228
  {
229
  "epoch": 1.3817243920412676,
230
+ "eval_loss": 0.9803809523582458,
231
+ "eval_runtime": 74.9936,
232
+ "eval_samples_per_second": 16.095,
233
+ "eval_steps_per_second": 2.014,
234
  "step": 7500
235
  },
236
  {
237
  "epoch": 1.4738393515106853,
238
+ "grad_norm": 0.0976635068655014,
239
  "learning_rate": 3.1577008106116435e-08,
240
+ "loss": 0.196,
241
  "step": 8000
242
  },
243
  {
244
  "epoch": 1.4738393515106853,
245
+ "eval_loss": 0.9711655378341675,
246
+ "eval_runtime": 74.3537,
247
+ "eval_samples_per_second": 16.233,
248
+ "eval_steps_per_second": 2.031,
249
  "step": 8000
250
  },
251
  {
252
  "epoch": 1.565954310980103,
253
+ "grad_norm": 46.64162826538086,
254
  "learning_rate": 3.042557111274871e-08,
255
+ "loss": 0.1946,
256
  "step": 8500
257
  },
258
  {
259
  "epoch": 1.565954310980103,
260
+ "eval_loss": 0.9621181488037109,
261
+ "eval_runtime": 75.7205,
262
+ "eval_samples_per_second": 15.94,
263
+ "eval_steps_per_second": 1.994,
264
  "step": 8500
265
  },
266
  {
267
  "epoch": 1.658069270449521,
268
+ "grad_norm": 0.05690698325634003,
269
  "learning_rate": 2.927413411938099e-08,
270
+ "loss": 0.2422,
271
  "step": 9000
272
  },
273
  {
274
  "epoch": 1.658069270449521,
275
+ "eval_loss": 0.9536784291267395,
276
+ "eval_runtime": 75.1745,
277
+ "eval_samples_per_second": 16.056,
278
+ "eval_steps_per_second": 2.009,
279
  "step": 9000
280
  }
281
  ],
checkpoint-9000/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:595bb6f39a89ec7030e395471c015665757cb5cacaafa6bba1613fd32b629686
3
  size 4847
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a29c2374e59ce3f57b34db9cb20d2d1bdc870c3dad128e9a4d462636c37e8b0
3
  size 4847