{ "checkpoint": "/public/home/winniechen/maniptrans/runs/handx_seq/handx_diffusion_x0norm_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": 101, "best_score": 0.2702867090702057, "selection_metric": "val_loss", "obs_dims": { "proprioception": 79, "privileged": 22, "target": 184, "last_action": 28 }, "obs_dim": 313, "action_dim": 28, "normalization": { "normalize": true, "obs_mean": [ 0.0952436551451683, -0.009415031410753727, 0.10264027118682861, 0.10573142021894455, 0.1803433895111084, -0.008514774963259697, 0.18269172310829163, 0.18365508317947388, 0.2618010640144348, 0.16608066856861115, 0.02162761799991131, 0.18216069042682648, 0.18358978629112244, 0.17237836122512817, -0.015642715618014336, 0.17602336406707764, 0.17724169790744781, 0.043983206152915955, 0.1670200079679489, 0.15985101461410522, 0.18348316848278046, 0.15706467628479004, 0.9937220811843872, 0.9985200762748718, 0.9929639101028442, 0.9924666881561279, 0.9835179448127747, 0.9985379576683044, 0.9830723404884338, 0.9828567504882812, 0.9659203290939331, 0.9857601523399353, 0.9971596002578735, 0.9829581379890442, 0.9826850295066833, 0.9843910336494446, 0.9986034035682678, 0.9837914109230042, 0.9835864305496216, 0.9967789649963379, 0.9853898286819458, 0.9865813851356506, 0.982226550579071, 0.9865478277206421, 0.0949554443359375, -0.009414969012141228, 0.1022946685552597, 0.1053469330072403, 0.17932210862636566, -0.008517003618180752, 0.1816285252571106, 0.18256936967372894, 0.2588196098804474, 0.1652478128671646, 0.0215448047965765, 0.18107609450817108, 0.18247942626476288, 0.17141476273536682, -0.01562916487455368, 0.17500633001327515, 0.17620635032653809, 0.04385775700211525, 0.16614864766597748, 0.15908151865005493, 0.18230782449245453, 0.15628452599048615, 0.0, 0.0, 0.0, 0.14385901391506195, 0.425918310880661, 0.3195565342903137, 0.6628227233886719, 0.028299279510974884, -0.04231749847531319, -0.01997886784374714, 0.4642455279827118, 0.3134498596191406, -0.11411713808774948, -0.005324739031493664, -0.006816457025706768, 0.0012877549743279815, 0.002557376166805625, 0.01799030788242817, -0.0030941551085561514, 0.01827973686158657, 0.017936356365680695, 0.0, 0.007644005585461855, -0.0077636828646063805, 0.007922005839645863, 0.00822428148239851, 0.02501649223268032, -0.004693011753261089, 0.021681809797883034, 0.022724496200680733, 0.00044409785186871886, -0.0005474574281834066, 0.0034559229388833046, -0.008890010416507721, 0.0018029273487627506, 0.0010117619531229138, -0.001323189353570342, -0.0005092014907859266, 0.14824940264225006, 0.4277472496032715, 0.32014843821525574, 0.6610589623451233, -0.004617097321897745, 0.0004904986708424985, 0.0005805635591968894, 0.9833521842956543, 0.0010167558211833239, -0.0013239731779322028, -0.0005023133708164096, 0.0010167568689212203, -0.0013239772524684668, -0.0005023131961934268, 0.0010127294808626175, -0.00129030947573483, -0.0005583541351370513, 0.001005804748274386, -0.0012544789351522923, -0.0005694699357263744, 0.0009956661378964782, -0.001214254880324006, -0.0005539834965020418, 0.0010133745381608605, -0.001323973061516881, -0.0006357521051540971, 0.001013376866467297, -0.0013239672407507896, -0.0006357518141157925, 0.0009911191882565618, -0.0012603553477674723, -0.0006679772632196546, 0.0009656506590545177, -0.0011863071704283357, -0.0006558148306794465, 0.0009383266442455351, -0.0011006221175193787, -0.000618424266576767, 0.0009830924682319164, -0.0012471856316551566, -0.0007358686998486519, 0.000993967056274414, -0.0012851940700784326, -0.000830993230920285, 0.0009939668234437704, -0.0012851882493123412, -0.000830993230920285, 0.0009765038848854601, -0.0012177641037851572, -0.0008526907186023891, 0.0009546198998577893, -0.0011406992562115192, -0.000816759595181793, 0.0009276768541894853, -0.0010517725022509694, -0.0007458801846951246, 0.0010054021840915084, -0.0013093991437926888, -0.0007321457960642874, 0.0010054020676761866, -0.00130940112285316, -0.0007321456796489656, 0.0009831995703279972, -0.0012433371739462018, -0.0007649928447790444, 0.0009552671108394861, -0.0011653489200398326, -0.0007542191888205707, 0.0009218008490279317, -0.0010742676677182317, -0.0007192263146862388, 0.0009992691921070218, -0.0012705966364592314, -0.00031518444302491844, 0.0010094373719766736, -0.0013021554332226515, -0.0003550891706254333, 0.0010055031161755323, -0.001238307449966669, -0.00010278530680807307, 0.0010055035818368196, -0.0012383091961964965, -0.00010278505214955658, 0.0010217326926067472, -0.001243004109710455, -0.00011843202810268849, 0.0010293828090652823, -0.0012229724088683724, -0.00014388498675543815, 0.010802571661770344, -0.015549628995358944, 0.00032347181695513427, 0.17909331619739532, 0.4687412679195404, 0.33615508675575256, 0.602735698223114, -0.03667610138654709, -0.0053223855793476105, -0.0012291927123442292, 0.9020772576332092, 0.011872115544974804, -0.015066142193973064, -0.0023922829423099756, 0.0118721267208457, -0.01506620179861784, -0.0023922843392938375, 0.011604099534451962, -0.01420511957257986, -0.004639853723347187, 0.011235728859901428, -0.013673043809831142, -0.005884566344320774, 0.010783546604216099, -0.0132643673568964, -0.0066540539264678955, 0.01122378371655941, -0.014441428706049919, -0.004093322902917862, 0.011223826557397842, -0.014441338367760181, -0.004093319643288851, 0.010637011379003525, -0.01298225112259388, -0.006057804450392723, 0.009887523017823696, -0.01173526793718338, -0.007031864486634731, 0.008989334106445312, -0.010524868033826351, -0.0075212218798696995, 0.00922988448292017, -0.01290720421820879, -0.005489319562911987, 0.009417271241545677, -0.013040200807154179, -0.006372714415192604, 0.00941727589815855, -0.013040116988122463, -0.006372716277837753, 0.00882906373590231, -0.011467375792562962, -0.008301449939608574, 0.008066831156611443, -0.010081946849822998, -0.009129698388278484, 0.00710753258317709, -0.00870242528617382, -0.009405317716300488, 0.010412070900201797, -0.013818172737956047, -0.005184386856853962, 0.010412068106234074, -0.013818209059536457, -0.005184385925531387, 0.009785957634449005, -0.012209952808916569, -0.007158895023167133, 0.008921255357563496, -0.010731338523328304, -0.008180559612810612, 0.007810310460627079, -0.009227749891579151, -0.008747725747525692, 0.011098423041403294, -0.01553611271083355, 0.001682989182882011, 0.011431134305894375, -0.01584516651928425, 0.0013910970883443952, 0.012403734028339386, -0.016053058207035065, 0.002713260240852833, 0.012403738684952259, -0.0160530898720026, 0.0027132618706673384, 0.013385052792727947, -0.016310056671500206, 0.0015878526028245687, 0.013931067660450935, -0.016124149784445763, 0.0005007712170481682, 0.00163837184663862, -0.0010326122865080833, 0.0005625621415674686, 0.026750145480036736, 0.03804921358823776, 0.04265772923827171, -0.6786400079727173, -0.025829434394836426, -0.8669411540031433, -0.8336186408996582, -0.5828378796577454, -0.02265218272805214, -0.7768170833587646, -0.7239329218864441, 0.9830508232116699, -0.5993273854255676, 0.06184905767440796, -0.7778398394584656, -0.7245330214500427, -0.5918521285057068, -0.04309089481830597, -0.7844624519348145, -0.7330986261367798, -0.7776706218719482, 0.4702238440513611, -0.2831260859966278, 0.5163207054138184, -0.8061503767967224 ], "obs_std": [ 0.05920194834470749, 0.05360560864210129, 0.05970413610339165, 0.06259719282388687, 0.02318476140499115, 0.05341646075248718, 0.024074353277683258, 0.025724101811647415, 1.0, 0.031162891536951065, 0.0722455307841301, 0.031707316637039185, 0.03213205561041832, 0.039911653846502304, 0.05050517991185188, 0.03911017253994942, 0.038876768201589584, 0.06721940636634827, 0.037457384169101715, 0.03675581514835358, 0.044711191207170486, 0.04798843711614609, 0.0048615350387990475, 0.0023326987866312265, 0.005492880940437317, 0.005939336493611336, 0.003874132176861167, 0.0018698282074183226, 0.003965221345424652, 0.004153145477175713, 1.0, 0.004280773922801018, 0.003856808412820101, 0.004649242851883173, 0.004663132131099701, 0.005125954281538725, 0.0016652023186907172, 0.005144920200109482, 0.005145875737071037, 0.004688414745032787, 0.0046485960483551025, 0.004484391771256924, 0.006029258947819471, 0.00487435283139348, 0.058953944593667984, 0.05351055786013603, 0.05940602347254753, 0.06226370111107826, 0.022851157933473587, 0.05334572494029999, 0.023736627772450447, 0.02537577413022518, 1.0, 0.030853640288114548, 0.07205818593502045, 0.03134823590517044, 0.03177415579557419, 0.03954581916332245, 0.05044696852564812, 0.03873682767152786, 0.03850127011537552, 0.06697964668273926, 0.03714326024055481, 0.036458857357501984, 0.0442567877471447, 0.047687336802482605, 1.0, 1.0, 1.0, 0.26602330803871155, 0.3275105357170105, 0.17311006784439087, 0.2201124131679535, 0.20044474303722382, 0.2581091821193695, 0.25801166892051697, 4.924596786499023, 7.988438606262207, 5.408953666687012, 0.2941558063030243, 0.2620483338832855, 0.2860102951526642, 0.29839131236076355, 0.31388747692108154, 0.30551281571388245, 0.31515195965766907, 0.3271336853504181, 1.0, 0.1823350340127945, 0.3521619141101837, 0.18656064569950104, 0.176376610994339, 0.38097622990608215, 0.28346574306488037, 0.3253062069416046, 0.3210950791835785, 0.4423409700393677, 0.12108424305915833, 0.2041032314300537, 0.41081467270851135, 0.24270935356616974, 0.006899653468281031, 0.008881524205207825, 0.008805342018604279, 0.263753741979599, 0.326436847448349, 0.17220079898834229, 0.22311556339263916, 0.04207708314061165, 0.026184339076280594, 0.02809724397957325, 0.17248491942882538, 0.006917153485119343, 0.006831647828221321, 0.0059961238875985146, 0.006917148362845182, 0.006831645499914885, 0.005996124353259802, 0.006909602787345648, 0.006919125095009804, 0.005411030724644661, 0.0069698127917945385, 0.007217324338853359, 0.0054262191988527775, 0.007091483101248741, 0.007560535799711943, 0.005771656986325979, 0.006817877758294344, 0.00717961136251688, 0.006116562522947788, 0.006817879155278206, 0.007179612293839455, 0.0061165629886090755, 0.0068649169988930225, 0.00718665262684226, 0.00545407272875309, 0.006954217795282602, 0.0074717034585773945, 0.005371200852096081, 0.007103240117430687, 0.007819443009793758, 0.005529215559363365, 0.0068919463083148, 0.008902233093976974, 0.006648004986345768, 0.007011485286056995, 0.00911321397870779, 0.007043267600238323, 0.007011479698121548, 0.00911321584135294, 0.0070432680658996105, 0.007192783523350954, 0.008671190589666367, 0.005924600176513195, 0.007353051565587521, 0.008636098355054855, 0.005433522630482912, 0.007540253922343254, 0.008750303648412228, 0.005327168386429548, 0.006833961699157953, 0.00792199932038784, 0.006499121431261301, 0.006833959836512804, 0.007921998389065266, 0.006499121896922588, 0.006921177264302969, 0.007720830850303173, 0.0057059722021222115, 0.0070262751542031765, 0.007912486791610718, 0.005513378418982029, 0.007175842300057411, 0.008225500583648682, 0.005583909805864096, 0.006901809945702553, 0.0077994647435843945, 0.007576304953545332, 0.006945409812033176, 0.0077102468349039555, 0.0076665752567350864, 0.007536168675869703, 0.0072046141140162945, 0.006601553875952959, 0.007536168675869703, 0.007204612251371145, 0.006601555272936821, 0.007951285690069199, 0.007726728916168213, 0.0065249004401266575, 0.008200298994779587, 0.008343582041561604, 0.0067316871136426926, 0.0652691200375557, 0.06622470915317535, 0.04892759397625923, 0.25418388843536377, 0.2996484935283661, 0.17163823544979095, 0.2967448830604553, 0.2106076031923294, 0.12751072645187378, 0.1095484122633934, 0.3350611627101898, 0.06856035441160202, 0.06890545040369034, 0.049959540367126465, 0.06856036186218262, 0.06890544295310974, 0.049959540367126465, 0.06941158324480057, 0.07127449661493301, 0.05126352608203888, 0.07006257772445679, 0.07331237941980362, 0.053156036883592606, 0.07072007656097412, 0.07517603039741516, 0.05531686544418335, 0.06750056147575378, 0.06880590319633484, 0.04836399480700493, 0.06750060617923737, 0.06880591064691544, 0.04836399853229523, 0.06830668449401855, 0.07158462703227997, 0.050096966326236725, 0.0687640979886055, 0.07386783510446548, 0.05196337401866913, 0.0691259428858757, 0.07588250935077667, 0.0538107194006443, 0.0655522495508194, 0.06913193315267563, 0.04679946228861809, 0.0655648484826088, 0.06950514018535614, 0.04692104831337929, 0.0655648484826088, 0.06950521469116211, 0.04692104458808899, 0.06638345867395401, 0.07204464077949524, 0.04877537116408348, 0.06688272207975388, 0.07416006177663803, 0.05061529576778412, 0.06725580990314484, 0.07600557804107666, 0.052369628101587296, 0.06642773002386093, 0.06875131279230118, 0.04728826880455017, 0.06642772257328033, 0.06875137239694595, 0.047288261353969574, 0.06721219420433044, 0.07143155485391617, 0.049346961081027985, 0.06760648638010025, 0.07368233054876328, 0.05137082189321518, 0.06783232092857361, 0.07568944245576859, 0.053302157670259476, 0.0666838064789772, 0.06665337830781937, 0.051213353872299194, 0.06683309376239777, 0.06666388362646103, 0.05092120170593262, 0.07042279094457626, 0.07043235003948212, 0.05708855763077736, 0.07042279094457626, 0.07043233513832092, 0.05708855763077736, 0.0726749449968338, 0.0734044760465622, 0.05984891206026077, 0.07407864928245544, 0.07569830864667892, 0.062152303755283356, 0.011775447055697441, 0.02019038237631321, 0.020082255825400352, 0.6339901685714722, 0.4912031888961792, 0.6474385261535645, 0.11180192232131958, 0.15264828503131866, 0.13260483741760254, 0.14110372960567474, 0.08038849383592606, 0.15146970748901367, 0.10518274456262589, 0.10113527625799179, 0.12908051908016205, 0.08636585623025894, 0.20603974163532257, 0.10836154967546463, 0.10564210265874863, 0.08944343030452728, 0.1439489722251892, 0.111920565366745, 0.11055395752191544, 0.12031196802854538, 0.12324045598506927, 0.053136009722948074, 0.144452303647995, 0.11916554719209671 ], "action_mean": [ 0.0017069753957912326, -0.0009585986263118684, 0.0005176332197152078, 0.025624632835388184, 0.033922385424375534, 0.035424381494522095, -0.6910957098007202, -0.027657141909003258, -0.882256805896759, -0.8483575582504272, -0.5928143262863159, -0.02468125708401203, -0.7901046276092529, -0.7362897396087646, 1.0, -0.6094473600387573, 0.06118907779455185, -0.7909987568855286, -0.7366766333580017, -0.6017321944236755, -0.045255813747644424, -0.7976285219192505, -0.7452515363693237, -0.7917129397392273, 0.4783249795436859, -0.28803086280822754, 0.5252812504768372, -0.8200786113739014 ], "action_std": [ 0.01182060968130827, 0.020237736403942108, 0.020106365904211998, 0.6420328617095947, 0.5012528896331787, 0.6524461507797241, 0.06839858740568161, 0.1538485884666443, 0.06875547766685486, 0.09003522247076035, 0.024705741554498672, 0.1527296006679535, 0.02577568031847477, 0.034674957394599915, 1.0, 0.035674113780260086, 0.2077256590127945, 0.03630285710096359, 0.046067576855421066, 0.04441961646080017, 0.14523029327392578, 0.043943341821432114, 0.05460977181792259, 0.06451387703418732, 0.10813111811876297, 0.03859848901629448, 0.12841075658798218, 0.055439915508031845 ] }, "history": [ { "epoch": 1, "train_loss": 1.5683873891830444, "val_loss": 0.8547106385231018, "val_action_mse": 0.07085013389587402 }, { "epoch": 2, "train_loss": 1.3626450300216675, "val_loss": 0.8061419129371643, "val_action_mse": 0.06562568247318268 }, { "epoch": 3, "train_loss": 1.2702672481536865, "val_loss": 0.7599486112594604, "val_action_mse": 0.05102473124861717 }, { "epoch": 4, "train_loss": 1.287641167640686, "val_loss": 0.7100340723991394, "val_action_mse": 0.05656600371003151 }, { "epoch": 5, "train_loss": 1.1127731800079346, "val_loss": 0.6559359431266785, "val_action_mse": 0.05887051299214363 }, { "epoch": 6, "train_loss": 1.0496689081192017, "val_loss": 0.6047946214675903, "val_action_mse": 0.060022637248039246 }, { "epoch": 7, "train_loss": 1.03534996509552, "val_loss": 0.5551183819770813, "val_action_mse": 0.05281166359782219 }, { "epoch": 8, "train_loss": 0.8900895118713379, "val_loss": 0.5115025639533997, "val_action_mse": 0.0526573620736599 }, { "epoch": 9, "train_loss": 0.8674135208129883, "val_loss": 0.47986382246017456, "val_action_mse": 0.048252373933792114 }, { "epoch": 10, "train_loss": 0.8281657099723816, "val_loss": 0.4611380100250244, "val_action_mse": 0.046049814671278 }, { "epoch": 11, "train_loss": 1.0160914659500122, "val_loss": 0.44646042585372925, "val_action_mse": 0.043080803006887436 }, { "epoch": 12, "train_loss": 0.8386415243148804, "val_loss": 0.4387780725955963, "val_action_mse": 0.04279310628771782 }, { "epoch": 13, "train_loss": 0.8127113580703735, "val_loss": 0.4335840940475464, "val_action_mse": 0.040091622620821 }, { "epoch": 14, "train_loss": 0.7632583379745483, "val_loss": 0.4322950541973114, "val_action_mse": 0.0400068499147892 }, { "epoch": 15, "train_loss": 0.6514544486999512, "val_loss": 0.4257607161998749, "val_action_mse": 0.042856499552726746 }, { "epoch": 16, "train_loss": 0.6691828966140747, "val_loss": 0.41557157039642334, "val_action_mse": 0.04104601964354515 }, { "epoch": 17, "train_loss": 0.737218976020813, "val_loss": 0.4028964638710022, "val_action_mse": 0.04038244113326073 }, { "epoch": 18, "train_loss": 0.9439409971237183, "val_loss": 0.3902783989906311, "val_action_mse": 0.04640429466962814 }, { "epoch": 19, "train_loss": 0.6243469715118408, "val_loss": 0.38080787658691406, "val_action_mse": 0.04179713875055313 }, { "epoch": 20, "train_loss": 0.5888206958770752, "val_loss": 0.3716150224208832, "val_action_mse": 0.045028671622276306 }, { "epoch": 21, "train_loss": 0.5820206999778748, "val_loss": 0.36431100964546204, "val_action_mse": 0.04400251433253288 }, { "epoch": 22, "train_loss": 0.6941290497779846, "val_loss": 0.35672256350517273, "val_action_mse": 0.040208201855421066 }, { "epoch": 23, "train_loss": 0.6965101957321167, "val_loss": 0.35181671380996704, "val_action_mse": 0.05337588116526604 }, { "epoch": 24, "train_loss": 0.5302689075469971, "val_loss": 0.34907251596450806, "val_action_mse": 0.055029381066560745 }, { "epoch": 25, "train_loss": 0.5772925615310669, "val_loss": 0.3480274975299835, "val_action_mse": 0.05890684574842453 }, { "epoch": 26, "train_loss": 0.675922155380249, "val_loss": 0.34407705068588257, "val_action_mse": 0.056258466094732285 }, { "epoch": 27, "train_loss": 0.6778231263160706, "val_loss": 0.3362743854522705, "val_action_mse": 0.05551532283425331 }, { "epoch": 28, "train_loss": 0.6648889780044556, "val_loss": 0.3300628066062927, "val_action_mse": 0.047202158719301224 }, { "epoch": 29, "train_loss": 0.493793785572052, "val_loss": 0.32488375902175903, "val_action_mse": 0.04546162113547325 }, { "epoch": 30, "train_loss": 0.5059396624565125, "val_loss": 0.32198160886764526, "val_action_mse": 0.041800569742918015 }, { "epoch": 31, "train_loss": 0.46726560592651367, "val_loss": 0.3207104206085205, "val_action_mse": 0.042003195732831955 }, { "epoch": 32, "train_loss": 0.6787493228912354, "val_loss": 0.31919553875923157, "val_action_mse": 0.04239851236343384 }, { "epoch": 33, "train_loss": 0.4858269691467285, "val_loss": 0.3182159960269928, "val_action_mse": 0.04233355075120926 }, { "epoch": 34, "train_loss": 0.9581775069236755, "val_loss": 0.32003989815711975, "val_action_mse": 0.04835449904203415 }, { "epoch": 35, "train_loss": 0.427217036485672, "val_loss": 0.3211383819580078, "val_action_mse": 0.04445268213748932 }, { "epoch": 36, "train_loss": 0.5524064302444458, "val_loss": 0.31949207186698914, "val_action_mse": 0.047406040132045746 }, { "epoch": 37, "train_loss": 0.6456826329231262, "val_loss": 0.3151682913303375, "val_action_mse": 0.04261060431599617 }, { "epoch": 38, "train_loss": 0.5609813928604126, "val_loss": 0.307475745677948, "val_action_mse": 0.04747913405299187 }, { "epoch": 39, "train_loss": 0.4680243730545044, "val_loss": 0.3012641668319702, "val_action_mse": 0.046535056084394455 }, { "epoch": 40, "train_loss": 0.5176399946212769, "val_loss": 0.29919707775115967, "val_action_mse": 0.040229566395282745 }, { "epoch": 41, "train_loss": 0.41160014271736145, "val_loss": 0.29684850573539734, "val_action_mse": 0.03877553716301918 }, { "epoch": 42, "train_loss": 0.4614565074443817, "val_loss": 0.2976647615432739, "val_action_mse": 0.03916215896606445 }, { "epoch": 43, "train_loss": 0.5306558012962341, "val_loss": 0.2992209494113922, "val_action_mse": 0.03977629914879799 }, { "epoch": 44, "train_loss": 0.36020734906196594, "val_loss": 0.30048155784606934, "val_action_mse": 0.04013802856206894 }, { "epoch": 45, "train_loss": 0.505221962928772, "val_loss": 0.30481037497520447, "val_action_mse": 0.04043792560696602 }, { "epoch": 46, "train_loss": 0.41825810074806213, "val_loss": 0.30928266048431396, "val_action_mse": 0.04140863195061684 }, { "epoch": 47, "train_loss": 0.4187264144420624, "val_loss": 0.31163784861564636, "val_action_mse": 0.04399072751402855 }, { "epoch": 48, "train_loss": 0.4423798620700836, "val_loss": 0.31144726276397705, "val_action_mse": 0.04214823991060257 }, { "epoch": 49, "train_loss": 0.4579353928565979, "val_loss": 0.3098594844341278, "val_action_mse": 0.04122261703014374 }, { "epoch": 50, "train_loss": 0.5290878415107727, "val_loss": 0.3063099980354309, "val_action_mse": 0.03995261713862419 }, { "epoch": 51, "train_loss": 0.3980328142642975, "val_loss": 0.30389460921287537, "val_action_mse": 0.040035173296928406 }, { "epoch": 52, "train_loss": 0.4131651222705841, "val_loss": 0.30114078521728516, "val_action_mse": 0.04091798886656761 }, { "epoch": 53, "train_loss": 0.5905667543411255, "val_loss": 0.3040955066680908, "val_action_mse": 0.048457883298397064 }, { "epoch": 54, "train_loss": 0.4732120633125305, "val_loss": 0.30666980147361755, "val_action_mse": 0.0405951552093029 }, { "epoch": 55, "train_loss": 0.42031848430633545, "val_loss": 0.31111595034599304, "val_action_mse": 0.0444488525390625 }, { "epoch": 56, "train_loss": 0.3935992121696472, "val_loss": 0.3128158450126648, "val_action_mse": 0.03957320749759674 }, { "epoch": 57, "train_loss": 0.49373018741607666, "val_loss": 0.3138057291507721, "val_action_mse": 0.04174182191491127 }, { "epoch": 58, "train_loss": 0.43276867270469666, "val_loss": 0.3140590488910675, "val_action_mse": 0.04152578487992287 }, { "epoch": 59, "train_loss": 0.42654600739479065, "val_loss": 0.3160848617553711, "val_action_mse": 0.04026682302355766 }, { "epoch": 60, "train_loss": 0.5189012289047241, "val_loss": 0.31576263904571533, "val_action_mse": 0.03997788578271866 }, { "epoch": 61, "train_loss": 0.31087160110473633, "val_loss": 0.3124179542064667, "val_action_mse": 0.04579045623540878 }, { "epoch": 62, "train_loss": 0.4905098080635071, "val_loss": 0.3096396327018738, "val_action_mse": 0.041269127279520035 }, { "epoch": 63, "train_loss": 0.4346162676811218, "val_loss": 0.30332741141319275, "val_action_mse": 0.03876931220293045 }, { "epoch": 64, "train_loss": 0.35898151993751526, "val_loss": 0.2991222143173218, "val_action_mse": 0.04104236140847206 }, { "epoch": 65, "train_loss": 0.37372222542762756, "val_loss": 0.2964091897010803, "val_action_mse": 0.0402684323489666 }, { "epoch": 66, "train_loss": 0.5471977591514587, "val_loss": 0.3008594810962677, "val_action_mse": 0.039330217987298965 }, { "epoch": 67, "train_loss": 0.661513090133667, "val_loss": 0.31205499172210693, "val_action_mse": 0.040590573102235794 }, { "epoch": 68, "train_loss": 0.4853987693786621, "val_loss": 0.3335364758968353, "val_action_mse": 0.0430564358830452 }, { "epoch": 69, "train_loss": 0.3196197748184204, "val_loss": 0.3503194749355316, "val_action_mse": 0.04080100357532501 }, { "epoch": 70, "train_loss": 0.40089789032936096, "val_loss": 0.3505185544490814, "val_action_mse": 0.042825814336538315 }, { "epoch": 71, "train_loss": 0.4889054298400879, "val_loss": 0.349252849817276, "val_action_mse": 0.043107498437166214 }, { "epoch": 72, "train_loss": 0.32643404603004456, "val_loss": 0.33794620633125305, "val_action_mse": 0.04153282195329666 }, { "epoch": 73, "train_loss": 0.39354798197746277, "val_loss": 0.32579097151756287, "val_action_mse": 0.04525475949048996 }, { "epoch": 74, "train_loss": 0.2967120409011841, "val_loss": 0.31759411096572876, "val_action_mse": 0.039782460778951645 }, { "epoch": 75, "train_loss": 0.32077524065971375, "val_loss": 0.31204912066459656, "val_action_mse": 0.03963645547628403 }, { "epoch": 76, "train_loss": 0.28973016142845154, "val_loss": 0.30920103192329407, "val_action_mse": 0.03955213353037834 }, { "epoch": 77, "train_loss": 0.3039146959781647, "val_loss": 0.3083365261554718, "val_action_mse": 0.038674164563417435 }, { "epoch": 78, "train_loss": 0.34194013476371765, "val_loss": 0.3114212155342102, "val_action_mse": 0.03829197585582733 }, { "epoch": 79, "train_loss": 0.2621977627277374, "val_loss": 0.315502405166626, "val_action_mse": 0.03877009451389313 }, { "epoch": 80, "train_loss": 0.3374316990375519, "val_loss": 0.3199392259120941, "val_action_mse": 0.0374060794711113 }, { "epoch": 81, "train_loss": 0.2985214591026306, "val_loss": 0.32018613815307617, "val_action_mse": 0.040952663868665695 }, { "epoch": 82, "train_loss": 0.2525595426559448, "val_loss": 0.3181775212287903, "val_action_mse": 0.04005755856633186 }, { "epoch": 83, "train_loss": 0.28014418482780457, "val_loss": 0.31304869055747986, "val_action_mse": 0.04537152498960495 }, { "epoch": 84, "train_loss": 0.368274062871933, "val_loss": 0.3045494556427002, "val_action_mse": 0.037141576409339905 }, { "epoch": 85, "train_loss": 0.3361856937408447, "val_loss": 0.2932811379432678, "val_action_mse": 0.03747938573360443 }, { "epoch": 86, "train_loss": 0.3784373998641968, "val_loss": 0.28348982334136963, "val_action_mse": 0.037323061376810074 }, { "epoch": 87, "train_loss": 0.32002630829811096, "val_loss": 0.27921342849731445, "val_action_mse": 0.04613462835550308 }, { "epoch": 88, "train_loss": 0.24770420789718628, "val_loss": 0.27690306305885315, "val_action_mse": 0.037851765751838684 }, { "epoch": 89, "train_loss": 0.22876834869384766, "val_loss": 0.2778785228729248, "val_action_mse": 0.0382651723921299 }, { "epoch": 90, "train_loss": 0.2881617844104767, "val_loss": 0.2815767526626587, "val_action_mse": 0.0409877635538578 }, { "epoch": 91, "train_loss": 0.3369377553462982, "val_loss": 0.2870665192604065, "val_action_mse": 0.04060212895274162 }, { "epoch": 92, "train_loss": 0.20841555297374725, "val_loss": 0.29209545254707336, "val_action_mse": 0.04041001945734024 }, { "epoch": 93, "train_loss": 0.37185749411582947, "val_loss": 0.29481881856918335, "val_action_mse": 0.048155754804611206 }, { "epoch": 94, "train_loss": 0.3752882778644562, "val_loss": 0.29453951120376587, "val_action_mse": 0.03800259903073311 }, { "epoch": 95, "train_loss": 0.3216675817966461, "val_loss": 0.29169899225234985, "val_action_mse": 0.04722736030817032 }, { "epoch": 96, "train_loss": 0.3045155107975006, "val_loss": 0.2875848114490509, "val_action_mse": 0.03985333815217018 }, { "epoch": 97, "train_loss": 0.36748525500297546, "val_loss": 0.282053142786026, "val_action_mse": 0.04575897008180618 }, { "epoch": 98, "train_loss": 0.2735157907009125, "val_loss": 0.27734410762786865, "val_action_mse": 0.03772974759340286 }, { "epoch": 99, "train_loss": 0.23492024838924408, "val_loss": 0.2736990749835968, "val_action_mse": 0.03917190060019493 }, { "epoch": 100, "train_loss": 0.2919411063194275, "val_loss": 0.2719235420227051, "val_action_mse": 0.04593394324183464 }, { "epoch": 101, "train_loss": 0.2526591122150421, "val_loss": 0.2702867090702057, "val_action_mse": 0.03912137448787689 }, { "epoch": 102, "train_loss": 0.3297446668148041, "val_loss": 0.2710546851158142, "val_action_mse": 0.03710981085896492 }, { "epoch": 103, "train_loss": 0.24895738065242767, "val_loss": 0.2733558416366577, "val_action_mse": 0.039484743028879166 }, { "epoch": 104, "train_loss": 0.3269806504249573, "val_loss": 0.2761986255645752, "val_action_mse": 0.036846309900283813 }, { "epoch": 105, "train_loss": 0.2239413857460022, "val_loss": 0.2795206308364868, "val_action_mse": 0.03795498237013817 }, { "epoch": 106, "train_loss": 0.2427031248807907, "val_loss": 0.28046831488609314, "val_action_mse": 0.03733767196536064 }, { "epoch": 107, "train_loss": 0.22596201300621033, "val_loss": 0.2812369465827942, "val_action_mse": 0.037854064255952835 }, { "epoch": 108, "train_loss": 0.31333667039871216, "val_loss": 0.28120872378349304, "val_action_mse": 0.04020741581916809 }, { "epoch": 109, "train_loss": 0.22430144250392914, "val_loss": 0.27967149019241333, "val_action_mse": 0.03688839450478554 }, { "epoch": 110, "train_loss": 0.2122674435377121, "val_loss": 0.2772729992866516, "val_action_mse": 0.03824545443058014 }, { "epoch": 111, "train_loss": 0.2739374041557312, "val_loss": 0.2766666114330292, "val_action_mse": 0.040212202817201614 }, { "epoch": 112, "train_loss": 0.3195199966430664, "val_loss": 0.2775694727897644, "val_action_mse": 0.040154214948415756 }, { "epoch": 113, "train_loss": 0.2165296971797943, "val_loss": 0.2776079773902893, "val_action_mse": 0.043140873312950134 }, { "epoch": 114, "train_loss": 0.28298959136009216, "val_loss": 0.2787620723247528, "val_action_mse": 0.03653280809521675 }, { "epoch": 115, "train_loss": 0.2899683713912964, "val_loss": 0.2798255681991577, "val_action_mse": 0.03962633013725281 }, { "epoch": 116, "train_loss": 0.22341185808181763, "val_loss": 0.2804270088672638, "val_action_mse": 0.03736409172415733 }, { "epoch": 117, "train_loss": 0.2512209117412567, "val_loss": 0.282813161611557, "val_action_mse": 0.04159807413816452 }, { "epoch": 118, "train_loss": 0.23138056695461273, "val_loss": 0.2854591906070709, "val_action_mse": 0.03867342695593834 }, { "epoch": 119, "train_loss": 0.29544341564178467, "val_loss": 0.2862624526023865, "val_action_mse": 0.03973246365785599 }, { "epoch": 120, "train_loss": 0.2906021177768707, "val_loss": 0.2861894369125366, "val_action_mse": 0.038252443075180054 } ], "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." } }