{
  "checkpoint": "/public/home/winniechen/maniptrans/runs/handx_seq/handx_diffusion_train_retarget_sharpa_rh_8clips.pth",
  "dataset": "/public/home/winniechen/maniptrans/data/handx_bc/handx_train_retarget_sharpa_rh_8clips.npz",
  "model_type": "diffusion",
  "best_epoch": 55,
  "best_val_action_mse": 0.42593449354171753,
  "obs_dims": {
    "proprioception": 79,
    "privileged": 22,
    "target": 184,
    "last_action": 28
  },
  "obs_dim": 313,
  "action_dim": 28,
  "history": [
    {
      "epoch": 1,
      "train_loss": 1.225028395652771,
      "val_loss": 0.11234702914953232,
      "val_action_mse": 1.3069968223571777
    },
    {
      "epoch": 2,
      "train_loss": 1.137717366218567,
      "val_loss": 0.05165449157357216,
      "val_action_mse": 1.3295199871063232
    },
    {
      "epoch": 3,
      "train_loss": 1.0596141815185547,
      "val_loss": 0.030436763539910316,
      "val_action_mse": 1.3932222127914429
    },
    {
      "epoch": 4,
      "train_loss": 1.0194625854492188,
      "val_loss": 0.02301706187427044,
      "val_action_mse": 1.3665437698364258
    },
    {
      "epoch": 5,
      "train_loss": 0.9919118285179138,
      "val_loss": 0.017177999019622803,
      "val_action_mse": 1.3639483451843262
    },
    {
      "epoch": 6,
      "train_loss": 0.9902920126914978,
      "val_loss": 0.014173110947012901,
      "val_action_mse": 1.3040562868118286
    },
    {
      "epoch": 7,
      "train_loss": 0.985162615776062,
      "val_loss": 0.014766921289265156,
      "val_action_mse": 1.2881280183792114
    },
    {
      "epoch": 8,
      "train_loss": 0.996105968952179,
      "val_loss": 0.012915444560348988,
      "val_action_mse": 1.2732199430465698
    },
    {
      "epoch": 9,
      "train_loss": 0.9501965045928955,
      "val_loss": 0.011687341146171093,
      "val_action_mse": 1.2355761528015137
    },
    {
      "epoch": 10,
      "train_loss": 0.9400526285171509,
      "val_loss": 0.012033187784254551,
      "val_action_mse": 1.227973222732544
    },
    {
      "epoch": 11,
      "train_loss": 0.946111798286438,
      "val_loss": 0.01411326415836811,
      "val_action_mse": 1.2248083353042603
    },
    {
      "epoch": 12,
      "train_loss": 0.9459391236305237,
      "val_loss": 0.015561603009700775,
      "val_action_mse": 1.2241318225860596
    },
    {
      "epoch": 13,
      "train_loss": 0.9266873598098755,
      "val_loss": 0.01379028894007206,
      "val_action_mse": 1.1233245134353638
    },
    {
      "epoch": 14,
      "train_loss": 0.9946807026863098,
      "val_loss": 0.012254899367690086,
      "val_action_mse": 1.1320284605026245
    },
    {
      "epoch": 15,
      "train_loss": 0.870001494884491,
      "val_loss": 0.012009692378342152,
      "val_action_mse": 1.0561249256134033
    },
    {
      "epoch": 16,
      "train_loss": 0.886614978313446,
      "val_loss": 0.012694476172327995,
      "val_action_mse": 0.974909245967865
    },
    {
      "epoch": 17,
      "train_loss": 0.8479912281036377,
      "val_loss": 0.020013516768813133,
      "val_action_mse": 0.9591075778007507
    },
    {
      "epoch": 18,
      "train_loss": 0.7984098196029663,
      "val_loss": 0.023025384172797203,
      "val_action_mse": 0.8246648907661438
    },
    {
      "epoch": 19,
      "train_loss": 0.7518520355224609,
      "val_loss": 0.028767641633749008,
      "val_action_mse": 0.8475245237350464
    },
    {
      "epoch": 20,
      "train_loss": 0.6489991545677185,
      "val_loss": 0.06688880920410156,
      "val_action_mse": 0.7997101545333862
    },
    {
      "epoch": 21,
      "train_loss": 0.6510717272758484,
      "val_loss": 0.09863957017660141,
      "val_action_mse": 0.8441164493560791
    },
    {
      "epoch": 22,
      "train_loss": 0.617274820804596,
      "val_loss": 0.20388376712799072,
      "val_action_mse": 0.756892740726471
    },
    {
      "epoch": 23,
      "train_loss": 0.6640934944152832,
      "val_loss": 0.21054163575172424,
      "val_action_mse": 0.751570463180542
    },
    {
      "epoch": 24,
      "train_loss": 0.543434202671051,
      "val_loss": 0.18172046542167664,
      "val_action_mse": 0.7282963395118713
    },
    {
      "epoch": 25,
      "train_loss": 0.7537526488304138,
      "val_loss": 0.15474233031272888,
      "val_action_mse": 0.7252967953681946
    },
    {
      "epoch": 26,
      "train_loss": 0.5202249884605408,
      "val_loss": 0.11865789443254471,
      "val_action_mse": 0.762821614742279
    },
    {
      "epoch": 27,
      "train_loss": 0.39882421493530273,
      "val_loss": 0.09440199285745621,
      "val_action_mse": 0.7000719904899597
    },
    {
      "epoch": 28,
      "train_loss": 0.4824810028076172,
      "val_loss": 0.1177225261926651,
      "val_action_mse": 0.6619753241539001
    },
    {
      "epoch": 29,
      "train_loss": 0.5672624707221985,
      "val_loss": 0.13026446104049683,
      "val_action_mse": 0.6369376182556152
    },
    {
      "epoch": 30,
      "train_loss": 0.28313788771629333,
      "val_loss": 0.163180872797966,
      "val_action_mse": 0.6810802817344666
    },
    {
      "epoch": 31,
      "train_loss": 0.44505590200424194,
      "val_loss": 0.1425127238035202,
      "val_action_mse": 0.6775518655776978
    },
    {
      "epoch": 32,
      "train_loss": 0.4339587688446045,
      "val_loss": 0.14272668957710266,
      "val_action_mse": 0.6351771950721741
    },
    {
      "epoch": 33,
      "train_loss": 0.467261403799057,
      "val_loss": 0.1373273879289627,
      "val_action_mse": 0.6016308665275574
    },
    {
      "epoch": 34,
      "train_loss": 0.5147853493690491,
      "val_loss": 0.13622722029685974,
      "val_action_mse": 0.6244176030158997
    },
    {
      "epoch": 35,
      "train_loss": 0.41999995708465576,
      "val_loss": 0.15181964635849,
      "val_action_mse": 0.6198785305023193
    },
    {
      "epoch": 36,
      "train_loss": 0.36957958340644836,
      "val_loss": 0.14044371247291565,
      "val_action_mse": 0.6519373655319214
    },
    {
      "epoch": 37,
      "train_loss": 0.4845064878463745,
      "val_loss": 0.09843187779188156,
      "val_action_mse": 0.621709942817688
    },
    {
      "epoch": 38,
      "train_loss": 0.3402957618236542,
      "val_loss": 0.1149682030081749,
      "val_action_mse": 0.5966364145278931
    },
    {
      "epoch": 39,
      "train_loss": 0.20650020241737366,
      "val_loss": 0.1379825323820114,
      "val_action_mse": 0.5720919966697693
    },
    {
      "epoch": 40,
      "train_loss": 0.2740054726600647,
      "val_loss": 0.10092467069625854,
      "val_action_mse": 0.5694532990455627
    },
    {
      "epoch": 41,
      "train_loss": 0.3148754835128784,
      "val_loss": 0.08790987730026245,
      "val_action_mse": 0.6134791374206543
    },
    {
      "epoch": 42,
      "train_loss": 0.46780461072921753,
      "val_loss": 0.10927682369947433,
      "val_action_mse": 0.5462494492530823
    },
    {
      "epoch": 43,
      "train_loss": 0.2361287623643875,
      "val_loss": 0.10699024051427841,
      "val_action_mse": 0.5813165903091431
    },
    {
      "epoch": 44,
      "train_loss": 0.23656955361366272,
      "val_loss": 0.12773488461971283,
      "val_action_mse": 0.49497923254966736
    },
    {
      "epoch": 45,
      "train_loss": 0.3536296784877777,
      "val_loss": 0.14367717504501343,
      "val_action_mse": 0.5560709238052368
    },
    {
      "epoch": 46,
      "train_loss": 0.3094886243343353,
      "val_loss": 0.15738871693611145,
      "val_action_mse": 0.5179444551467896
    },
    {
      "epoch": 47,
      "train_loss": 0.30475640296936035,
      "val_loss": 0.16269312798976898,
      "val_action_mse": 0.5533431172370911
    },
    {
      "epoch": 48,
      "train_loss": 0.6201184391975403,
      "val_loss": 0.0955285057425499,
      "val_action_mse": 0.5536470413208008
    },
    {
      "epoch": 49,
      "train_loss": 0.6499500870704651,
      "val_loss": 0.07548889517784119,
      "val_action_mse": 0.5053880214691162
    },
    {
      "epoch": 50,
      "train_loss": 0.32918041944503784,
      "val_loss": 0.11273704469203949,
      "val_action_mse": 0.5130437016487122
    },
    {
      "epoch": 51,
      "train_loss": 0.36807313561439514,
      "val_loss": 0.09166351705789566,
      "val_action_mse": 0.5131313800811768
    },
    {
      "epoch": 52,
      "train_loss": 0.4341641068458557,
      "val_loss": 0.07583204656839371,
      "val_action_mse": 0.5121503472328186
    },
    {
      "epoch": 53,
      "train_loss": 0.28429755568504333,
      "val_loss": 0.05484766513109207,
      "val_action_mse": 0.46526527404785156
    },
    {
      "epoch": 54,
      "train_loss": 0.1827351450920105,
      "val_loss": 0.08446325361728668,
      "val_action_mse": 0.5235853791236877
    },
    {
      "epoch": 55,
      "train_loss": 0.21780157089233398,
      "val_loss": 0.1049148216843605,
      "val_action_mse": 0.42593449354171753
    },
    {
      "epoch": 56,
      "train_loss": 0.21971629559993744,
      "val_loss": 0.11875956505537033,
      "val_action_mse": 0.5470485687255859
    },
    {
      "epoch": 57,
      "train_loss": 0.3924177289009094,
      "val_loss": 0.11781518161296844,
      "val_action_mse": 0.4452790319919586
    },
    {
      "epoch": 58,
      "train_loss": 0.183973029255867,
      "val_loss": 0.11041709780693054,
      "val_action_mse": 0.44857028126716614
    },
    {
      "epoch": 59,
      "train_loss": 0.2920917570590973,
      "val_loss": 0.12673251330852509,
      "val_action_mse": 0.4882045388221741
    },
    {
      "epoch": 60,
      "train_loss": 0.14983810484409332,
      "val_loss": 0.1270756870508194,
      "val_action_mse": 0.4986453950405121
    },
    {
      "epoch": 61,
      "train_loss": 0.3323284685611725,
      "val_loss": 0.08973699063062668,
      "val_action_mse": 0.4494093656539917
    },
    {
      "epoch": 62,
      "train_loss": 0.30340346693992615,
      "val_loss": 0.06415016949176788,
      "val_action_mse": 0.48177069425582886
    },
    {
      "epoch": 63,
      "train_loss": 0.23527072370052338,
      "val_loss": 0.09321953356266022,
      "val_action_mse": 0.4733276665210724
    },
    {
      "epoch": 64,
      "train_loss": 0.4615941047668457,
      "val_loss": 0.09106287360191345,
      "val_action_mse": 0.49004095792770386
    },
    {
      "epoch": 65,
      "train_loss": 0.20810222625732422,
      "val_loss": 0.09063920378684998,
      "val_action_mse": 0.5143473148345947
    },
    {
      "epoch": 66,
      "train_loss": 0.22592514753341675,
      "val_loss": 0.09284403920173645,
      "val_action_mse": 0.49505287408828735
    },
    {
      "epoch": 67,
      "train_loss": 0.2481665313243866,
      "val_loss": 0.09635401517152786,
      "val_action_mse": 0.5329849123954773
    },
    {
      "epoch": 68,
      "train_loss": 0.30395814776420593,
      "val_loss": 0.12847012281417847,
      "val_action_mse": 0.5241488814353943
    },
    {
      "epoch": 69,
      "train_loss": 0.17520123720169067,
      "val_loss": 0.18987977504730225,
      "val_action_mse": 0.507979154586792
    },
    {
      "epoch": 70,
      "train_loss": 0.26298636198043823,
      "val_loss": 0.15374034643173218,
      "val_action_mse": 0.47933560609817505
    },
    {
      "epoch": 71,
      "train_loss": 0.21759122610092163,
      "val_loss": 0.12028006464242935,
      "val_action_mse": 0.48771315813064575
    },
    {
      "epoch": 72,
      "train_loss": 0.14932438731193542,
      "val_loss": 0.12648716568946838,
      "val_action_mse": 0.45724886655807495
    },
    {
      "epoch": 73,
      "train_loss": 0.3017114996910095,
      "val_loss": 0.14949920773506165,
      "val_action_mse": 0.5094899535179138
    },
    {
      "epoch": 74,
      "train_loss": 0.22263455390930176,
      "val_loss": 0.12437821179628372,
      "val_action_mse": 0.4832560420036316
    },
    {
      "epoch": 75,
      "train_loss": 0.5674774050712585,
      "val_loss": 0.08618242293596268,
      "val_action_mse": 0.5171247124671936
    },
    {
      "epoch": 76,
      "train_loss": 0.13941065967082977,
      "val_loss": 0.10740279406309128,
      "val_action_mse": 0.45408564805984497
    },
    {
      "epoch": 77,
      "train_loss": 0.4969983994960785,
      "val_loss": 0.09431881457567215,
      "val_action_mse": 0.538411557674408
    },
    {
      "epoch": 78,
      "train_loss": 0.2487073689699173,
      "val_loss": 0.0855531319975853,
      "val_action_mse": 0.4965178072452545
    },
    {
      "epoch": 79,
      "train_loss": 0.14941878616809845,
      "val_loss": 0.08607187122106552,
      "val_action_mse": 0.5130510926246643
    },
    {
      "epoch": 80,
      "train_loss": 0.13209772109985352,
      "val_loss": 0.08642018586397171,
      "val_action_mse": 0.562786877155304
    },
    {
      "epoch": 81,
      "train_loss": 0.24012668430805206,
      "val_loss": 0.0829787626862526,
      "val_action_mse": 0.4676757752895355
    },
    {
      "epoch": 82,
      "train_loss": 0.21844379603862762,
      "val_loss": 0.07422208786010742,
      "val_action_mse": 0.4632870554924011
    },
    {
      "epoch": 83,
      "train_loss": 0.24086512625217438,
      "val_loss": 0.10795268416404724,
      "val_action_mse": 0.5466957092285156
    },
    {
      "epoch": 84,
      "train_loss": 0.17570064961910248,
      "val_loss": 0.13029170036315918,
      "val_action_mse": 0.47307223081588745
    },
    {
      "epoch": 85,
      "train_loss": 0.19598782062530518,
      "val_loss": 0.1201162338256836,
      "val_action_mse": 0.46898287534713745
    },
    {
      "epoch": 86,
      "train_loss": 0.17500336468219757,
      "val_loss": 0.1071813553571701,
      "val_action_mse": 0.4533085525035858
    },
    {
      "epoch": 87,
      "train_loss": 0.2993457019329071,
      "val_loss": 0.08398623764514923,
      "val_action_mse": 0.5165582895278931
    },
    {
      "epoch": 88,
      "train_loss": 0.1911824494600296,
      "val_loss": 0.09009258449077606,
      "val_action_mse": 0.503616452217102
    },
    {
      "epoch": 89,
      "train_loss": 0.22411713004112244,
      "val_loss": 0.09833342581987381,
      "val_action_mse": 0.4796442687511444
    },
    {
      "epoch": 90,
      "train_loss": 0.23988264799118042,
      "val_loss": 0.09825634211301804,
      "val_action_mse": 0.5013913512229919
    },
    {
      "epoch": 91,
      "train_loss": 0.15476307272911072,
      "val_loss": 0.10194797813892365,
      "val_action_mse": 0.4566631317138672
    },
    {
      "epoch": 92,
      "train_loss": 0.1540805697441101,
      "val_loss": 0.09060966223478317,
      "val_action_mse": 0.48257821798324585
    },
    {
      "epoch": 93,
      "train_loss": 0.16343426704406738,
      "val_loss": 0.09829318523406982,
      "val_action_mse": 0.5179023146629333
    },
    {
      "epoch": 94,
      "train_loss": 0.18914055824279785,
      "val_loss": 0.09627379477024078,
      "val_action_mse": 0.4796859622001648
    },
    {
      "epoch": 95,
      "train_loss": 0.16516029834747314,
      "val_loss": 0.0940607339143753,
      "val_action_mse": 0.5197756886482239
    },
    {
      "epoch": 96,
      "train_loss": 0.19147683680057526,
      "val_loss": 0.10966607928276062,
      "val_action_mse": 0.4272536337375641
    },
    {
      "epoch": 97,
      "train_loss": 0.20265516638755798,
      "val_loss": 0.13320603966712952,
      "val_action_mse": 0.4616169333457947
    },
    {
      "epoch": 98,
      "train_loss": 0.28691136837005615,
      "val_loss": 0.10907495021820068,
      "val_action_mse": 0.5303149819374084
    },
    {
      "epoch": 99,
      "train_loss": 0.1495584100484848,
      "val_loss": 0.08774635195732117,
      "val_action_mse": 0.5024663805961609
    },
    {
      "epoch": 100,
      "train_loss": 0.14334328472614288,
      "val_loss": 0.08015959709882736,
      "val_action_mse": 0.5454575419425964
    },
    {
      "epoch": 101,
      "train_loss": 0.2650372087955475,
      "val_loss": 0.07966701686382294,
      "val_action_mse": 0.5087618827819824
    },
    {
      "epoch": 102,
      "train_loss": 0.24847005307674408,
      "val_loss": 0.07247164845466614,
      "val_action_mse": 0.45921894907951355
    },
    {
      "epoch": 103,
      "train_loss": 0.13823767006397247,
      "val_loss": 0.08925852924585342,
      "val_action_mse": 0.4867765009403229
    },
    {
      "epoch": 104,
      "train_loss": 0.15648117661476135,
      "val_loss": 0.11446696519851685,
      "val_action_mse": 0.4361768960952759
    },
    {
      "epoch": 105,
      "train_loss": 0.2709158658981323,
      "val_loss": 0.09997939318418503,
      "val_action_mse": 0.485729455947876
    },
    {
      "epoch": 106,
      "train_loss": 0.29123419523239136,
      "val_loss": 0.07884787768125534,
      "val_action_mse": 0.4595548212528229
    },
    {
      "epoch": 107,
      "train_loss": 0.17335090041160583,
      "val_loss": 0.07917046546936035,
      "val_action_mse": 0.5108315944671631
    },
    {
      "epoch": 108,
      "train_loss": 0.16448616981506348,
      "val_loss": 0.08136655390262604,
      "val_action_mse": 0.47722136974334717
    },
    {
      "epoch": 109,
      "train_loss": 0.2695733308792114,
      "val_loss": 0.06743410229682922,
      "val_action_mse": 0.5169175863265991
    },
    {
      "epoch": 110,
      "train_loss": 0.21141597628593445,
      "val_loss": 0.08010764420032501,
      "val_action_mse": 0.5235785245895386
    },
    {
      "epoch": 111,
      "train_loss": 0.13492488861083984,
      "val_loss": 0.10119358450174332,
      "val_action_mse": 0.526080846786499
    },
    {
      "epoch": 112,
      "train_loss": 0.1263774335384369,
      "val_loss": 0.0966203510761261,
      "val_action_mse": 0.4889354705810547
    },
    {
      "epoch": 113,
      "train_loss": 0.19807375967502594,
      "val_loss": 0.0779036432504654,
      "val_action_mse": 0.5005959868431091
    },
    {
      "epoch": 114,
      "train_loss": 0.19317112863063812,
      "val_loss": 0.09022592008113861,
      "val_action_mse": 0.47393476963043213
    },
    {
      "epoch": 115,
      "train_loss": 0.23475539684295654,
      "val_loss": 0.09475289285182953,
      "val_action_mse": 0.47935473918914795
    },
    {
      "epoch": 116,
      "train_loss": 0.3398853838443756,
      "val_loss": 0.08227640390396118,
      "val_action_mse": 0.4692298471927643
    },
    {
      "epoch": 117,
      "train_loss": 0.14010311663150787,
      "val_loss": 0.08744736015796661,
      "val_action_mse": 0.5407394766807556
    },
    {
      "epoch": 118,
      "train_loss": 0.12566208839416504,
      "val_loss": 0.09501863270998001,
      "val_action_mse": 0.4981845021247864
    },
    {
      "epoch": 119,
      "train_loss": 0.20248153805732727,
      "val_loss": 0.08820420503616333,
      "val_action_mse": 0.5274531841278076
    },
    {
      "epoch": 120,
      "train_loss": 0.24669380486011505,
      "val_loss": 0.07642220705747604,
      "val_action_mse": 0.4499325752258301
    }
  ],
  "handx_reference": {
    "project": "https://handx-project.github.io/",
    "repo": "https://github.com/handx-project/HandX",
    "adaptation": "HandX-style autoregressive/diffusion sequence modeling with ManipTrans Stage-1 observation/action I/O."
  }
}