| { |
| "config": { |
| "dataPath": "data/paired/real-paired-clamp.jsonl", |
| "outputDir": "models/wiflow-real", |
| "totalEpochs": 50, |
| "batchSize": 32, |
| "lr": 0.00005, |
| "skipContrastive": true, |
| "evalSplit": 0.2, |
| "verbose": false, |
| "contrastiveRatio": 0.16666666666666666, |
| "supervisedRatio": 0.6666666666666666, |
| "refinementRatio": 0.16666666666666666, |
| "curriculumStages": [ |
| 0.9, |
| 0.7, |
| 0.5, |
| 0 |
| ], |
| "timeSteps": 20, |
| "numKeypoints": 17, |
| "momentum": 0.9, |
| "boneWeight": 0.3, |
| "temporalWeight": 0.1, |
| "inputDim": 35, |
| "contrastiveEpochs": 0, |
| "supervisedEpochs": 33, |
| "refinementEpochs": 17 |
| }, |
| "phases": [ |
| { |
| "phase": "supervised", |
| "epochs": [ |
| { |
| "epoch": 8, |
| "stage": 2, |
| "confThreshold": 0.7, |
| "loss": 0.1202345175039164, |
| "evalLoss": 0.1202650439646902, |
| "pck20": 0.8542199488491047, |
| "lr": 0.00004309335095262676, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 9, |
| "stage": 2, |
| "confThreshold": 0.7, |
| "loss": 0.12202338882917359, |
| "evalLoss": 0.12768733107391242, |
| "pck20": 0.6990622335890878, |
| "lr": 0.00004137151834863213, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 10, |
| "stage": 2, |
| "confThreshold": 0.7, |
| "loss": 0.12825024591293, |
| "evalLoss": 0.1270527974255793, |
| "pck20": 0.7263427109974423, |
| "lr": 0.000039501422739279956, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 11, |
| "stage": 2, |
| "confThreshold": 0.7, |
| "loss": 0.12936392720518736, |
| "evalLoss": 0.13037437100569574, |
| "pck20": 0.757885763000853, |
| "lr": 0.000037500000000000003, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 12, |
| "stage": 2, |
| "confThreshold": 0.7, |
| "loss": 0.12725472830309995, |
| "evalLoss": 0.12384917381806493, |
| "pck20": 0.7374254049445861, |
| "lr": 0.000035385375325047166, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 13, |
| "stage": 2, |
| "confThreshold": 0.7, |
| "loss": 0.12385725997187584, |
| "evalLoss": 0.12461516379265465, |
| "pck20": 0.7433930093776641, |
| "lr": 0.000033176699082935545, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 14, |
| "stage": 2, |
| "confThreshold": 0.7, |
| "loss": 0.121912843072227, |
| "evalLoss": 0.11932654898819652, |
| "pck20": 0.7706734867860185, |
| "lr": 0.000030893973387735687, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 15, |
| "stage": 2, |
| "confThreshold": 0.7, |
| "loss": 0.11553358296530863, |
| "evalLoss": 0.11448958151882806, |
| "pck20": 0.9036658141517468, |
| "lr": 0.000028557870956832132, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 16, |
| "stage": 3, |
| "confThreshold": 0.5, |
| "loss": 0.11394043133921662, |
| "evalLoss": 0.11232644030163798, |
| "pck20": 0.9096334185848243, |
| "lr": 0.000026189547895593562, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 17, |
| "stage": 3, |
| "confThreshold": 0.5, |
| "loss": 0.10948012598944228, |
| "evalLoss": 0.10736237566429192, |
| "pck20": 0.9190110826939465, |
| "lr": 0.000023810452104406444, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 18, |
| "stage": 3, |
| "confThreshold": 0.5, |
| "loss": 0.10622061844175851, |
| "evalLoss": 0.10554791605589925, |
| "pck20": 0.9224211423699906, |
| "lr": 0.000021442129043167874, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 19, |
| "stage": 3, |
| "confThreshold": 0.5, |
| "loss": 0.10454447340293044, |
| "evalLoss": 0.10382043899664628, |
| "pck20": 0.9215686274509797, |
| "lr": 0.000019106026612264326, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 20, |
| "stage": 3, |
| "confThreshold": 0.5, |
| "loss": 0.10241404779846577, |
| "evalLoss": 0.10137100661047628, |
| "pck20": 0.9232736572890018, |
| "lr": 0.00001682330091706446, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 21, |
| "stage": 3, |
| "confThreshold": 0.5, |
| "loss": 0.10052735078244243, |
| "evalLoss": 0.10080157734640532, |
| "pck20": 0.9232736572890018, |
| "lr": 0.000014614624674952837, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 22, |
| "stage": 3, |
| "confThreshold": 0.5, |
| "loss": 0.09994398953615607, |
| "evalLoss": 0.10015142701439284, |
| "pck20": 0.9232736572890018, |
| "lr": 0.000012500000000000006, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 23, |
| "stage": 3, |
| "confThreshold": 0.5, |
| "loss": 0.09929149190664616, |
| "evalLoss": 0.09950004892959943, |
| "pck20": 0.9232736572890018, |
| "lr": 0.00001049857726072005, |
| "trainSamples": 276 |
| }, |
| { |
| "epoch": 24, |
| "stage": 4, |
| "confThreshold": 0, |
| "loss": 0.09857283315945391, |
| "evalLoss": 0.09854543363902651, |
| "pck20": 0.9232736572890018, |
| "lr": 0.000008628481651367876, |
| "trainSamples": 552 |
| }, |
| { |
| "epoch": 25, |
| "stage": 4, |
| "confThreshold": 0, |
| "loss": 0.09782671456921876, |
| "evalLoss": 0.09771453065823558, |
| "pck20": 0.9232736572890018, |
| "lr": 0.000006906649047373246, |
| "trainSamples": 552 |
| }, |
| { |
| "epoch": 26, |
| "stage": 4, |
| "confThreshold": 0, |
| "loss": 0.09710556690724673, |
| "evalLoss": 0.0969643238344358, |
| "pck20": 0.9232736572890018, |
| "lr": 0.000005348672631430313, |
| "trainSamples": 552 |
| }, |
| { |
| "epoch": 27, |
| "stage": 4, |
| "confThreshold": 0, |
| "loss": 0.09644318116959444, |
| "evalLoss": 0.09656814571783083, |
| "pck20": 0.9232736572890018, |
| "lr": 0.000003968661679220473, |
| "trainSamples": 552 |
| }, |
| { |
| "epoch": 28, |
| "stage": 4, |
| "confThreshold": 0, |
| "loss": 0.09587479137308834, |
| "evalLoss": 0.09621587546104406, |
| "pck20": 0.9232736572890018, |
| "lr": 0.000002779113783626916, |
| "trainSamples": 552 |
| }, |
| { |
| "epoch": 29, |
| "stage": 4, |
| "confThreshold": 0, |
| "loss": 0.095646920219553, |
| "evalLoss": 0.09603035061275683, |
| "pck20": 0.9232736572890018, |
| "lr": 0.000001790801674598186, |
| "trainSamples": 552 |
| }, |
| { |
| "epoch": 30, |
| "stage": 4, |
| "confThreshold": 0, |
| "loss": 0.09563373023370147, |
| "evalLoss": 0.09590667483554177, |
| "pck20": 0.9232736572890018, |
| "lr": 0.0000010126756596375686, |
| "trainSamples": 552 |
| }, |
| { |
| "epoch": 31, |
| "stage": 4, |
| "confThreshold": 0, |
| "loss": 0.09565320996677595, |
| "evalLoss": 0.09583995214951894, |
| "pck20": 0.9232736572890018, |
| "lr": 4.5178256843233513e-7, |
| "trainSamples": 552 |
| }, |
| { |
| "epoch": 32, |
| "stage": 4, |
| "confThreshold": 0, |
| "loss": 0.09559600900155292, |
| "evalLoss": 0.09582890515717067, |
| "pck20": 0.9232736572890018, |
| "lr": 1.1320193567288529e-7, |
| "trainSamples": 552 |
| } |
| ] |
| }, |
| { |
| "phase": "refinement", |
| "epochs": [ |
| { |
| "epoch": 33, |
| "loss": 0.09701607626199084, |
| "boneLoss": 0.008328167800411033, |
| "temporalLoss": 0, |
| "evalLoss": 0.09398452057857512, |
| "pck20": 0.9224211423699906, |
| "lr": 0.000025 |
| }, |
| { |
| "epoch": 34, |
| "loss": 0.09559415324560026, |
| "boneLoss": 0.007928355392524818, |
| "temporalLoss": 0, |
| "evalLoss": 0.09261685967463618, |
| "pck20": 0.9232736572890018, |
| "lr": 0.000024787163746048776 |
| }, |
| { |
| "epoch": 35, |
| "loss": 0.0937338397275952, |
| "boneLoss": 0.007887900375719051, |
| "temporalLoss": 0, |
| "evalLoss": 0.09073041961606622, |
| "pck20": 0.9224211423699906, |
| "lr": 0.000024155902867554447 |
| }, |
| { |
| "epoch": 36, |
| "loss": 0.09201024094364278, |
| "boneLoss": 0.00787968505949037, |
| "temporalLoss": 0, |
| "evalLoss": 0.08904968648766556, |
| "pck20": 0.9224211423699906, |
| "lr": 0.00002312771419662018 |
| }, |
| { |
| "epoch": 37, |
| "loss": 0.0900738158246732, |
| "boneLoss": 0.007898852512024901, |
| "temporalLoss": 0, |
| "evalLoss": 0.08739649689835581, |
| "pck20": 0.9232736572890018, |
| "lr": 0.00002173761146525824 |
| }, |
| { |
| "epoch": 38, |
| "loss": 0.08872982797507926, |
| "boneLoss": 0.008049209921715138, |
| "temporalLoss": 0, |
| "evalLoss": 0.08634493550993017, |
| "pck20": 0.9275362318840571, |
| "lr": 0.000020032932954740708 |
| }, |
| { |
| "epoch": 39, |
| "loss": 0.08807447949516382, |
| "boneLoss": 0.008001218330457414, |
| "temporalLoss": 0, |
| "evalLoss": 0.08558730437074917, |
| "pck20": 0.9292412617220792, |
| "lr": 0.000018071729447206733 |
| }, |
| { |
| "epoch": 40, |
| "loss": 0.08706284437260242, |
| "boneLoss": 0.007840473818814284, |
| "temporalLoss": 0, |
| "evalLoss": 0.08471010713466626, |
| "pck20": 0.927536231884057, |
| "lr": 0.000015920787375901038 |
| }, |
| { |
| "epoch": 41, |
| "loss": 0.08604835426163508, |
| "boneLoss": 0.007663920377313184, |
| "temporalLoss": 0, |
| "evalLoss": 0.08400298512163402, |
| "pck20": 0.927536231884057, |
| "lr": 0.000013653354493291276 |
| }, |
| { |
| "epoch": 42, |
| "loss": 0.08560569613850501, |
| "boneLoss": 0.007651553561183805, |
| "temporalLoss": 0, |
| "evalLoss": 0.08322107553620108, |
| "pck20": 0.928388746803068, |
| "lr": 0.000011346645506708726 |
| }, |
| { |
| "epoch": 43, |
| "loss": 0.08482218876088875, |
| "boneLoss": 0.0076320660048834164, |
| "temporalLoss": 0, |
| "evalLoss": 0.08260210233693983, |
| "pck20": 0.9292412617220792, |
| "lr": 0.000009079212624098965 |
| }, |
| { |
| "epoch": 44, |
| "loss": 0.08466924374919721, |
| "boneLoss": 0.007591246856100821, |
| "temporalLoss": 0, |
| "evalLoss": 0.08220475184589006, |
| "pck20": 0.9292412617220792, |
| "lr": 0.000006928270552793279 |
| }, |
| { |
| "epoch": 45, |
| "loss": 0.08372166327234049, |
| "boneLoss": 0.007556008480447692, |
| "temporalLoss": 0, |
| "evalLoss": 0.08194887303484837, |
| "pck20": 0.9292412617220792, |
| "lr": 0.000004967067045259296 |
| }, |
| { |
| "epoch": 46, |
| "loss": 0.08351445528236756, |
| "boneLoss": 0.007548495414354707, |
| "temporalLoss": 0, |
| "evalLoss": 0.08170149165016606, |
| "pck20": 0.9292412617220792, |
| "lr": 0.0000032623885347417586 |
| }, |
| { |
| "epoch": 47, |
| "loss": 0.08341810004822017, |
| "boneLoss": 0.007546424241410138, |
| "temporalLoss": 0, |
| "evalLoss": 0.08158904121000707, |
| "pck20": 0.9292412617220792, |
| "lr": 0.0000018722858033798252 |
| }, |
| { |
| "epoch": 48, |
| "loss": 0.08368925729745821, |
| "boneLoss": 0.007544776057062225, |
| "temporalLoss": 0, |
| "evalLoss": 0.08154689865199782, |
| "pck20": 0.9292412617220792, |
| "lr": 8.440971324455538e-7 |
| }, |
| { |
| "epoch": 49, |
| "loss": 0.08322168186510166, |
| "boneLoss": 0.007544423146452779, |
| "temporalLoss": 0, |
| "evalLoss": 0.0815356110051266, |
| "pck20": 0.9292412617220792, |
| "lr": 2.128362539512277e-7 |
| } |
| ] |
| } |
| ] |
| } |
