{
  "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
        }
      ]
    }
  ]
}