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