ruv commited on
Commit
e04e86e
·
verified ·
1 Parent(s): 2271a56

Upload wiflow-v1/training-log.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. wiflow-v1/training-log.json +445 -0
wiflow-v1/training-log.json ADDED
@@ -0,0 +1,445 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "config": {
3
+ "dataPath": "data/paired/real-paired-clamp.jsonl",
4
+ "outputDir": "models/wiflow-real",
5
+ "totalEpochs": 50,
6
+ "batchSize": 32,
7
+ "lr": 0.00005,
8
+ "skipContrastive": true,
9
+ "evalSplit": 0.2,
10
+ "verbose": false,
11
+ "contrastiveRatio": 0.16666666666666666,
12
+ "supervisedRatio": 0.6666666666666666,
13
+ "refinementRatio": 0.16666666666666666,
14
+ "curriculumStages": [
15
+ 0.9,
16
+ 0.7,
17
+ 0.5,
18
+ 0
19
+ ],
20
+ "timeSteps": 20,
21
+ "numKeypoints": 17,
22
+ "momentum": 0.9,
23
+ "boneWeight": 0.3,
24
+ "temporalWeight": 0.1,
25
+ "inputDim": 35,
26
+ "contrastiveEpochs": 0,
27
+ "supervisedEpochs": 33,
28
+ "refinementEpochs": 17
29
+ },
30
+ "phases": [
31
+ {
32
+ "phase": "supervised",
33
+ "epochs": [
34
+ {
35
+ "epoch": 8,
36
+ "stage": 2,
37
+ "confThreshold": 0.7,
38
+ "loss": 0.1202345175039164,
39
+ "evalLoss": 0.1202650439646902,
40
+ "pck20": 0.8542199488491047,
41
+ "lr": 0.00004309335095262676,
42
+ "trainSamples": 276
43
+ },
44
+ {
45
+ "epoch": 9,
46
+ "stage": 2,
47
+ "confThreshold": 0.7,
48
+ "loss": 0.12202338882917359,
49
+ "evalLoss": 0.12768733107391242,
50
+ "pck20": 0.6990622335890878,
51
+ "lr": 0.00004137151834863213,
52
+ "trainSamples": 276
53
+ },
54
+ {
55
+ "epoch": 10,
56
+ "stage": 2,
57
+ "confThreshold": 0.7,
58
+ "loss": 0.12825024591293,
59
+ "evalLoss": 0.1270527974255793,
60
+ "pck20": 0.7263427109974423,
61
+ "lr": 0.000039501422739279956,
62
+ "trainSamples": 276
63
+ },
64
+ {
65
+ "epoch": 11,
66
+ "stage": 2,
67
+ "confThreshold": 0.7,
68
+ "loss": 0.12936392720518736,
69
+ "evalLoss": 0.13037437100569574,
70
+ "pck20": 0.757885763000853,
71
+ "lr": 0.000037500000000000003,
72
+ "trainSamples": 276
73
+ },
74
+ {
75
+ "epoch": 12,
76
+ "stage": 2,
77
+ "confThreshold": 0.7,
78
+ "loss": 0.12725472830309995,
79
+ "evalLoss": 0.12384917381806493,
80
+ "pck20": 0.7374254049445861,
81
+ "lr": 0.000035385375325047166,
82
+ "trainSamples": 276
83
+ },
84
+ {
85
+ "epoch": 13,
86
+ "stage": 2,
87
+ "confThreshold": 0.7,
88
+ "loss": 0.12385725997187584,
89
+ "evalLoss": 0.12461516379265465,
90
+ "pck20": 0.7433930093776641,
91
+ "lr": 0.000033176699082935545,
92
+ "trainSamples": 276
93
+ },
94
+ {
95
+ "epoch": 14,
96
+ "stage": 2,
97
+ "confThreshold": 0.7,
98
+ "loss": 0.121912843072227,
99
+ "evalLoss": 0.11932654898819652,
100
+ "pck20": 0.7706734867860185,
101
+ "lr": 0.000030893973387735687,
102
+ "trainSamples": 276
103
+ },
104
+ {
105
+ "epoch": 15,
106
+ "stage": 2,
107
+ "confThreshold": 0.7,
108
+ "loss": 0.11553358296530863,
109
+ "evalLoss": 0.11448958151882806,
110
+ "pck20": 0.9036658141517468,
111
+ "lr": 0.000028557870956832132,
112
+ "trainSamples": 276
113
+ },
114
+ {
115
+ "epoch": 16,
116
+ "stage": 3,
117
+ "confThreshold": 0.5,
118
+ "loss": 0.11394043133921662,
119
+ "evalLoss": 0.11232644030163798,
120
+ "pck20": 0.9096334185848243,
121
+ "lr": 0.000026189547895593562,
122
+ "trainSamples": 276
123
+ },
124
+ {
125
+ "epoch": 17,
126
+ "stage": 3,
127
+ "confThreshold": 0.5,
128
+ "loss": 0.10948012598944228,
129
+ "evalLoss": 0.10736237566429192,
130
+ "pck20": 0.9190110826939465,
131
+ "lr": 0.000023810452104406444,
132
+ "trainSamples": 276
133
+ },
134
+ {
135
+ "epoch": 18,
136
+ "stage": 3,
137
+ "confThreshold": 0.5,
138
+ "loss": 0.10622061844175851,
139
+ "evalLoss": 0.10554791605589925,
140
+ "pck20": 0.9224211423699906,
141
+ "lr": 0.000021442129043167874,
142
+ "trainSamples": 276
143
+ },
144
+ {
145
+ "epoch": 19,
146
+ "stage": 3,
147
+ "confThreshold": 0.5,
148
+ "loss": 0.10454447340293044,
149
+ "evalLoss": 0.10382043899664628,
150
+ "pck20": 0.9215686274509797,
151
+ "lr": 0.000019106026612264326,
152
+ "trainSamples": 276
153
+ },
154
+ {
155
+ "epoch": 20,
156
+ "stage": 3,
157
+ "confThreshold": 0.5,
158
+ "loss": 0.10241404779846577,
159
+ "evalLoss": 0.10137100661047628,
160
+ "pck20": 0.9232736572890018,
161
+ "lr": 0.00001682330091706446,
162
+ "trainSamples": 276
163
+ },
164
+ {
165
+ "epoch": 21,
166
+ "stage": 3,
167
+ "confThreshold": 0.5,
168
+ "loss": 0.10052735078244243,
169
+ "evalLoss": 0.10080157734640532,
170
+ "pck20": 0.9232736572890018,
171
+ "lr": 0.000014614624674952837,
172
+ "trainSamples": 276
173
+ },
174
+ {
175
+ "epoch": 22,
176
+ "stage": 3,
177
+ "confThreshold": 0.5,
178
+ "loss": 0.09994398953615607,
179
+ "evalLoss": 0.10015142701439284,
180
+ "pck20": 0.9232736572890018,
181
+ "lr": 0.000012500000000000006,
182
+ "trainSamples": 276
183
+ },
184
+ {
185
+ "epoch": 23,
186
+ "stage": 3,
187
+ "confThreshold": 0.5,
188
+ "loss": 0.09929149190664616,
189
+ "evalLoss": 0.09950004892959943,
190
+ "pck20": 0.9232736572890018,
191
+ "lr": 0.00001049857726072005,
192
+ "trainSamples": 276
193
+ },
194
+ {
195
+ "epoch": 24,
196
+ "stage": 4,
197
+ "confThreshold": 0,
198
+ "loss": 0.09857283315945391,
199
+ "evalLoss": 0.09854543363902651,
200
+ "pck20": 0.9232736572890018,
201
+ "lr": 0.000008628481651367876,
202
+ "trainSamples": 552
203
+ },
204
+ {
205
+ "epoch": 25,
206
+ "stage": 4,
207
+ "confThreshold": 0,
208
+ "loss": 0.09782671456921876,
209
+ "evalLoss": 0.09771453065823558,
210
+ "pck20": 0.9232736572890018,
211
+ "lr": 0.000006906649047373246,
212
+ "trainSamples": 552
213
+ },
214
+ {
215
+ "epoch": 26,
216
+ "stage": 4,
217
+ "confThreshold": 0,
218
+ "loss": 0.09710556690724673,
219
+ "evalLoss": 0.0969643238344358,
220
+ "pck20": 0.9232736572890018,
221
+ "lr": 0.000005348672631430313,
222
+ "trainSamples": 552
223
+ },
224
+ {
225
+ "epoch": 27,
226
+ "stage": 4,
227
+ "confThreshold": 0,
228
+ "loss": 0.09644318116959444,
229
+ "evalLoss": 0.09656814571783083,
230
+ "pck20": 0.9232736572890018,
231
+ "lr": 0.000003968661679220473,
232
+ "trainSamples": 552
233
+ },
234
+ {
235
+ "epoch": 28,
236
+ "stage": 4,
237
+ "confThreshold": 0,
238
+ "loss": 0.09587479137308834,
239
+ "evalLoss": 0.09621587546104406,
240
+ "pck20": 0.9232736572890018,
241
+ "lr": 0.000002779113783626916,
242
+ "trainSamples": 552
243
+ },
244
+ {
245
+ "epoch": 29,
246
+ "stage": 4,
247
+ "confThreshold": 0,
248
+ "loss": 0.095646920219553,
249
+ "evalLoss": 0.09603035061275683,
250
+ "pck20": 0.9232736572890018,
251
+ "lr": 0.000001790801674598186,
252
+ "trainSamples": 552
253
+ },
254
+ {
255
+ "epoch": 30,
256
+ "stage": 4,
257
+ "confThreshold": 0,
258
+ "loss": 0.09563373023370147,
259
+ "evalLoss": 0.09590667483554177,
260
+ "pck20": 0.9232736572890018,
261
+ "lr": 0.0000010126756596375686,
262
+ "trainSamples": 552
263
+ },
264
+ {
265
+ "epoch": 31,
266
+ "stage": 4,
267
+ "confThreshold": 0,
268
+ "loss": 0.09565320996677595,
269
+ "evalLoss": 0.09583995214951894,
270
+ "pck20": 0.9232736572890018,
271
+ "lr": 4.5178256843233513e-7,
272
+ "trainSamples": 552
273
+ },
274
+ {
275
+ "epoch": 32,
276
+ "stage": 4,
277
+ "confThreshold": 0,
278
+ "loss": 0.09559600900155292,
279
+ "evalLoss": 0.09582890515717067,
280
+ "pck20": 0.9232736572890018,
281
+ "lr": 1.1320193567288529e-7,
282
+ "trainSamples": 552
283
+ }
284
+ ]
285
+ },
286
+ {
287
+ "phase": "refinement",
288
+ "epochs": [
289
+ {
290
+ "epoch": 33,
291
+ "loss": 0.09701607626199084,
292
+ "boneLoss": 0.008328167800411033,
293
+ "temporalLoss": 0,
294
+ "evalLoss": 0.09398452057857512,
295
+ "pck20": 0.9224211423699906,
296
+ "lr": 0.000025
297
+ },
298
+ {
299
+ "epoch": 34,
300
+ "loss": 0.09559415324560026,
301
+ "boneLoss": 0.007928355392524818,
302
+ "temporalLoss": 0,
303
+ "evalLoss": 0.09261685967463618,
304
+ "pck20": 0.9232736572890018,
305
+ "lr": 0.000024787163746048776
306
+ },
307
+ {
308
+ "epoch": 35,
309
+ "loss": 0.0937338397275952,
310
+ "boneLoss": 0.007887900375719051,
311
+ "temporalLoss": 0,
312
+ "evalLoss": 0.09073041961606622,
313
+ "pck20": 0.9224211423699906,
314
+ "lr": 0.000024155902867554447
315
+ },
316
+ {
317
+ "epoch": 36,
318
+ "loss": 0.09201024094364278,
319
+ "boneLoss": 0.00787968505949037,
320
+ "temporalLoss": 0,
321
+ "evalLoss": 0.08904968648766556,
322
+ "pck20": 0.9224211423699906,
323
+ "lr": 0.00002312771419662018
324
+ },
325
+ {
326
+ "epoch": 37,
327
+ "loss": 0.0900738158246732,
328
+ "boneLoss": 0.007898852512024901,
329
+ "temporalLoss": 0,
330
+ "evalLoss": 0.08739649689835581,
331
+ "pck20": 0.9232736572890018,
332
+ "lr": 0.00002173761146525824
333
+ },
334
+ {
335
+ "epoch": 38,
336
+ "loss": 0.08872982797507926,
337
+ "boneLoss": 0.008049209921715138,
338
+ "temporalLoss": 0,
339
+ "evalLoss": 0.08634493550993017,
340
+ "pck20": 0.9275362318840571,
341
+ "lr": 0.000020032932954740708
342
+ },
343
+ {
344
+ "epoch": 39,
345
+ "loss": 0.08807447949516382,
346
+ "boneLoss": 0.008001218330457414,
347
+ "temporalLoss": 0,
348
+ "evalLoss": 0.08558730437074917,
349
+ "pck20": 0.9292412617220792,
350
+ "lr": 0.000018071729447206733
351
+ },
352
+ {
353
+ "epoch": 40,
354
+ "loss": 0.08706284437260242,
355
+ "boneLoss": 0.007840473818814284,
356
+ "temporalLoss": 0,
357
+ "evalLoss": 0.08471010713466626,
358
+ "pck20": 0.927536231884057,
359
+ "lr": 0.000015920787375901038
360
+ },
361
+ {
362
+ "epoch": 41,
363
+ "loss": 0.08604835426163508,
364
+ "boneLoss": 0.007663920377313184,
365
+ "temporalLoss": 0,
366
+ "evalLoss": 0.08400298512163402,
367
+ "pck20": 0.927536231884057,
368
+ "lr": 0.000013653354493291276
369
+ },
370
+ {
371
+ "epoch": 42,
372
+ "loss": 0.08560569613850501,
373
+ "boneLoss": 0.007651553561183805,
374
+ "temporalLoss": 0,
375
+ "evalLoss": 0.08322107553620108,
376
+ "pck20": 0.928388746803068,
377
+ "lr": 0.000011346645506708726
378
+ },
379
+ {
380
+ "epoch": 43,
381
+ "loss": 0.08482218876088875,
382
+ "boneLoss": 0.0076320660048834164,
383
+ "temporalLoss": 0,
384
+ "evalLoss": 0.08260210233693983,
385
+ "pck20": 0.9292412617220792,
386
+ "lr": 0.000009079212624098965
387
+ },
388
+ {
389
+ "epoch": 44,
390
+ "loss": 0.08466924374919721,
391
+ "boneLoss": 0.007591246856100821,
392
+ "temporalLoss": 0,
393
+ "evalLoss": 0.08220475184589006,
394
+ "pck20": 0.9292412617220792,
395
+ "lr": 0.000006928270552793279
396
+ },
397
+ {
398
+ "epoch": 45,
399
+ "loss": 0.08372166327234049,
400
+ "boneLoss": 0.007556008480447692,
401
+ "temporalLoss": 0,
402
+ "evalLoss": 0.08194887303484837,
403
+ "pck20": 0.9292412617220792,
404
+ "lr": 0.000004967067045259296
405
+ },
406
+ {
407
+ "epoch": 46,
408
+ "loss": 0.08351445528236756,
409
+ "boneLoss": 0.007548495414354707,
410
+ "temporalLoss": 0,
411
+ "evalLoss": 0.08170149165016606,
412
+ "pck20": 0.9292412617220792,
413
+ "lr": 0.0000032623885347417586
414
+ },
415
+ {
416
+ "epoch": 47,
417
+ "loss": 0.08341810004822017,
418
+ "boneLoss": 0.007546424241410138,
419
+ "temporalLoss": 0,
420
+ "evalLoss": 0.08158904121000707,
421
+ "pck20": 0.9292412617220792,
422
+ "lr": 0.0000018722858033798252
423
+ },
424
+ {
425
+ "epoch": 48,
426
+ "loss": 0.08368925729745821,
427
+ "boneLoss": 0.007544776057062225,
428
+ "temporalLoss": 0,
429
+ "evalLoss": 0.08154689865199782,
430
+ "pck20": 0.9292412617220792,
431
+ "lr": 8.440971324455538e-7
432
+ },
433
+ {
434
+ "epoch": 49,
435
+ "loss": 0.08322168186510166,
436
+ "boneLoss": 0.007544423146452779,
437
+ "temporalLoss": 0,
438
+ "evalLoss": 0.0815356110051266,
439
+ "pck20": 0.9292412617220792,
440
+ "lr": 2.128362539512277e-7
441
+ }
442
+ ]
443
+ }
444
+ ]
445
+ }