|
{ |
|
"arch_specifier": "no-align+fused-gelu-mlp", |
|
"architectures": [ |
|
"OpenVLAForActionPrediction" |
|
], |
|
"auto_map": { |
|
"AutoConfig": "configuration_prismatic.OpenVLAConfig", |
|
"AutoModelForVision2Seq": "modeling_prismatic.OpenVLAForActionPrediction" |
|
}, |
|
"hf_llm_id": "meta-llama/Llama-2-7b-hf", |
|
"image_resize_strategy": "resize-naive", |
|
"image_sizes": [ |
|
224, |
|
224 |
|
], |
|
"llm_backbone_id": "llama2-7b-pure", |
|
"llm_max_length": 2048, |
|
"model_type": "openvla", |
|
"n_action_bins": 256, |
|
"norm_stats": { |
|
"austin_buds_dataset_converted_externally_to_rlds": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
1.0, |
|
1.0, |
|
1.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
1.0 |
|
], |
|
"mean": [ |
|
-0.07678354531526566, |
|
0.0036849044263362885, |
|
0.05644911900162697, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.3510494828224182 |
|
], |
|
"min": [ |
|
-1.0, |
|
-1.0, |
|
-1.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
-1.0, |
|
-0.9599999785423279, |
|
-0.8714285492897034, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
1.0, |
|
0.8600000143051147, |
|
1.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
1.0 |
|
], |
|
"std": [ |
|
0.6367740631103516, |
|
0.37889179587364197, |
|
0.47796326875686646, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.47721168398857117 |
|
] |
|
}, |
|
"num_trajectories": 50, |
|
"num_transitions": 34112, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"austin_sailor_dataset_converted_externally_to_rlds": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
1.0, |
|
1.0, |
|
1.0, |
|
0.0, |
|
0.0, |
|
0.375, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.011825348250567913, |
|
0.006461074110120535, |
|
0.06023626774549484, |
|
0.0, |
|
0.0, |
|
0.0016465914668515325, |
|
0.5260950326919556 |
|
], |
|
"min": [ |
|
-1.0, |
|
-1.0, |
|
-1.0, |
|
0.0, |
|
0.0, |
|
-0.375, |
|
0.0 |
|
], |
|
"q01": [ |
|
-1.0, |
|
-0.9828571677207947, |
|
-0.6000000238418579, |
|
0.0, |
|
0.0, |
|
-0.17249999940395355, |
|
0.0 |
|
], |
|
"q99": [ |
|
1.0, |
|
0.9457142949104309, |
|
1.0, |
|
0.0, |
|
0.0, |
|
0.17892856895923615, |
|
1.0 |
|
], |
|
"std": [ |
|
0.46348899602890015, |
|
0.41240179538726807, |
|
0.411862850189209, |
|
0.0, |
|
0.0, |
|
0.0578610822558403, |
|
0.49894046783447266 |
|
] |
|
}, |
|
"num_trajectories": 240, |
|
"num_transitions": 353094, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"austin_sirius_dataset_converted_externally_to_rlds": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
1.0002285242080688, |
|
0.960608720779419, |
|
1.105179786682129, |
|
0.0, |
|
0.0, |
|
0.341785728931427, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.07747682929039001, |
|
0.03195561468601227, |
|
0.04244732856750488, |
|
0.0, |
|
0.0, |
|
-0.01603456400334835, |
|
0.43260177969932556 |
|
], |
|
"min": [ |
|
-1.0183025598526, |
|
-0.9800000190734863, |
|
-0.9774575233459473, |
|
0.0, |
|
0.0, |
|
-0.34607142210006714, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.780905865430832, |
|
-0.5667179036140442, |
|
-0.5254343223571777, |
|
0.0, |
|
0.0, |
|
-0.28495091378688814, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.9569637751579284, |
|
0.6971374487876891, |
|
0.8124888157844541, |
|
0.0, |
|
0.0, |
|
0.1971428543329239, |
|
1.0 |
|
], |
|
"std": [ |
|
0.3906329572200775, |
|
0.2998155355453491, |
|
0.2782271206378937, |
|
0.0, |
|
0.0, |
|
0.08120622485876083, |
|
0.49528297781944275 |
|
] |
|
}, |
|
"num_trajectories": 559, |
|
"num_transitions": 279939, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"bc_z": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.2165454924106598, |
|
0.1251407265663147, |
|
0.10772687941789627, |
|
0.33544227480888367, |
|
0.28117990493774414, |
|
0.40614867210388184, |
|
1.0 |
|
], |
|
"mean": [ |
|
-0.009958467446267605, |
|
0.0008958321413956583, |
|
0.004995597992092371, |
|
0.00029755113064311445, |
|
-0.008735382929444313, |
|
-0.030693737789988518, |
|
0.8344562649726868 |
|
], |
|
"min": [ |
|
-0.1677047461271286, |
|
-0.14630407094955444, |
|
-0.10066790133714676, |
|
-0.29421567916870117, |
|
-0.32101404666900635, |
|
-0.4635624885559082, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.09220654994249344, |
|
-0.06456145539879798, |
|
-0.049121275544166565, |
|
-0.11594625547528267, |
|
-0.14152548640966414, |
|
-0.2251061636209488, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.07628866866230968, |
|
0.058019736707210584, |
|
0.052540797740221024, |
|
0.11740604028105736, |
|
0.11703975558280955, |
|
0.16729306846857078, |
|
1.0 |
|
], |
|
"std": [ |
|
0.03053455986082554, |
|
0.0231423731893301, |
|
0.020641816779971123, |
|
0.04155943542718887, |
|
0.046427831053733826, |
|
0.0769818127155304, |
|
0.3610210120677948 |
|
] |
|
}, |
|
"num_trajectories": 43264, |
|
"num_transitions": 6015535, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"berkeley_autolab_ur5": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.019999999552965164, |
|
0.019999999552965164, |
|
0.019999999552965164, |
|
0.06666667014360428, |
|
0.06666667014360428, |
|
0.06666667014360428, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.0005683620693162084, |
|
0.001217700308188796, |
|
-0.0005296372692100704, |
|
0.00021029810886830091, |
|
6.0695128922816366e-05, |
|
0.001204986940138042, |
|
0.6298308372497559 |
|
], |
|
"min": [ |
|
-0.019999999552965164, |
|
-0.019999999552965164, |
|
-0.019999999552965164, |
|
-0.06666667014360428, |
|
-0.06666667014360428, |
|
-0.06666667014360428, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.019999999552965164, |
|
-0.019999999552965164, |
|
-0.019999999552965164, |
|
-0.02628571353852749, |
|
-0.06666667014360428, |
|
-0.03847619146108627, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.019999999552965164, |
|
0.019999999552965164, |
|
0.019999999552965164, |
|
0.031809523701667786, |
|
0.06666667014360428, |
|
0.036571428179740906, |
|
1.0 |
|
], |
|
"std": [ |
|
0.0115329809486866, |
|
0.007990492507815361, |
|
0.009577835910022259, |
|
0.009432995691895485, |
|
0.016427582129836082, |
|
0.011053967289626598, |
|
0.48267969489097595 |
|
] |
|
}, |
|
"num_trajectories": 1000, |
|
"num_transitions": 97939, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"berkeley_cable_routing": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.9633283019065857, |
|
1.0, |
|
1.0, |
|
0.0, |
|
0.0, |
|
1.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
-0.07139874249696732, |
|
0.023609008640050888, |
|
0.10241943597793579, |
|
0.0, |
|
0.0, |
|
0.049671024084091187, |
|
0.0 |
|
], |
|
"min": [ |
|
-0.9809081554412842, |
|
-0.9554349184036255, |
|
-0.9994775056838989, |
|
0.0, |
|
0.0, |
|
-1.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.5534318816661835, |
|
-0.4797285574674606, |
|
-0.5314934802055359, |
|
0.0, |
|
0.0, |
|
-0.8855219376087189, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.42652835428714786, |
|
0.5000944086909298, |
|
0.639823433756829, |
|
0.0, |
|
0.0, |
|
0.984243879914284, |
|
0.0 |
|
], |
|
"std": [ |
|
0.1815500408411026, |
|
0.1810990273952484, |
|
0.21220779418945312, |
|
0.0, |
|
0.0, |
|
0.3475511968135834, |
|
0.0 |
|
] |
|
}, |
|
"num_trajectories": 1647, |
|
"num_transitions": 42328, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"berkeley_fanuc_manipulation": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.009999999776482582, |
|
0.009999999776482582, |
|
0.009999999776482582, |
|
0.03490658476948738, |
|
0.03490658476948738, |
|
0.03490658476948738, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.0007744057802483439, |
|
-0.00031240080716088414, |
|
-0.0015001941937953234, |
|
-0.0007515158504247665, |
|
-0.00015832878125365824, |
|
0.00014327642566058785, |
|
0.699295699596405 |
|
], |
|
"min": [ |
|
-0.009999999776482582, |
|
-0.009999999776482582, |
|
-0.009999999776482582, |
|
-0.03490658476948738, |
|
-0.03490658476948738, |
|
-0.03490658476948738, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.009999999776482582, |
|
-0.009999999776482582, |
|
-0.009999999776482582, |
|
-0.03490658476948738, |
|
0.0, |
|
-0.03490658476948738, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.009999999776482582, |
|
0.009999999776482582, |
|
0.009999999776482582, |
|
0.03490658476948738, |
|
0.0, |
|
0.03490658476948738, |
|
1.0 |
|
], |
|
"std": [ |
|
0.0034070091787725687, |
|
0.0049921851605176926, |
|
0.005344334989786148, |
|
0.00759894959628582, |
|
0.004081866703927517, |
|
0.008568956516683102, |
|
0.4586937427520752 |
|
] |
|
}, |
|
"num_trajectories": 415, |
|
"num_transitions": 62613, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"bridge_orig": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.41691166162490845, |
|
0.25864794850349426, |
|
0.21218234300613403, |
|
3.122201919555664, |
|
1.8618112802505493, |
|
6.280478477478027, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.0002334194869035855, |
|
0.00013004911306779832, |
|
-0.00012762474943883717, |
|
-0.0001556558854645118, |
|
-0.0004039328487124294, |
|
0.00023557482927571982, |
|
0.5764579176902771 |
|
], |
|
"min": [ |
|
-0.4007510244846344, |
|
-0.13874775171279907, |
|
-0.22553899884223938, |
|
-3.2010786533355713, |
|
-1.8618112802505493, |
|
-6.279075622558594, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.02872725307941437, |
|
-0.04170349963009357, |
|
-0.026093858778476715, |
|
-0.08092105075716972, |
|
-0.09288699507713317, |
|
-0.20718276381492615, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.028309678435325586, |
|
0.040855254605412394, |
|
0.040161586627364146, |
|
0.08192047759890528, |
|
0.07792850524187081, |
|
0.20382574498653397, |
|
1.0 |
|
], |
|
"std": [ |
|
0.009765930473804474, |
|
0.013689135201275349, |
|
0.012667362578213215, |
|
0.028534092009067535, |
|
0.030637972056865692, |
|
0.07691419124603271, |
|
0.4973701536655426 |
|
] |
|
}, |
|
"num_trajectories": 60064, |
|
"num_transitions": 2135463, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"cmu_stretch": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.02338407188653946, |
|
0.0, |
|
0.023404927924275398, |
|
0.0, |
|
0.0, |
|
0.0, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.00036304505192674696, |
|
0.0, |
|
0.0016466958913952112, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.3987048268318176 |
|
], |
|
"min": [ |
|
-0.019353797659277916, |
|
0.0, |
|
-0.02019215188920498, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.011175686959177256, |
|
0.0, |
|
-0.0032206363626755773, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.014501785952597848, |
|
0.0, |
|
0.015056106168776728, |
|
0.0, |
|
0.0, |
|
0.0, |
|
1.0 |
|
], |
|
"std": [ |
|
0.004081828519701958, |
|
0.0, |
|
0.0037743328139185905, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.48963725566864014 |
|
] |
|
}, |
|
"num_trajectories": 135, |
|
"num_transitions": 25016, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"dlr_edan_shared_control_converted_externally_to_rlds": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.18991442024707794, |
|
0.0739002525806427, |
|
0.18064819276332855, |
|
0.0866486132144928, |
|
0.13464981317520142, |
|
0.16910280287265778, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.006647810339927673, |
|
-0.0007657372043468058, |
|
0.006522852927446365, |
|
0.0011679717572405934, |
|
-0.006395625416189432, |
|
-0.011902998201549053, |
|
0.6985887289047241 |
|
], |
|
"min": [ |
|
-0.10054297000169754, |
|
-0.08427435159683228, |
|
-0.13533438742160797, |
|
-0.17556548118591309, |
|
-0.18485672771930695, |
|
-0.2680685818195343, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.02987122368067503, |
|
-0.06013262912631035, |
|
-0.08286409199237824, |
|
-0.05924444157630205, |
|
-0.15986866518855095, |
|
-0.15636983573436739, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.08832092039287087, |
|
0.042126184627413736, |
|
0.11311905644834042, |
|
0.0643695573508739, |
|
0.03941855944693088, |
|
0.156646853685379, |
|
1.0 |
|
], |
|
"std": [ |
|
0.021393608301877975, |
|
0.01814231649041176, |
|
0.03374375030398369, |
|
0.01743541844189167, |
|
0.03394376486539841, |
|
0.04641875624656677, |
|
0.4588589072227478 |
|
] |
|
}, |
|
"num_trajectories": 104, |
|
"num_transitions": 8928, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"dobbe": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
38.590423583984375, |
|
17.932697296142578, |
|
4.843764305114746, |
|
1.4372116327285767, |
|
0.4340403974056244, |
|
1.2057193517684937, |
|
0.9998947381973267 |
|
], |
|
"mean": [ |
|
-0.0001120665911003016, |
|
0.0011229600058868527, |
|
-0.00010194431524723768, |
|
-7.371398532995954e-05, |
|
-0.00067531579406932, |
|
-5.6643435527803376e-05, |
|
0.6318281888961792 |
|
], |
|
"min": [ |
|
-5.700923442840576, |
|
-21.605947494506836, |
|
-123.72489929199219, |
|
-1.7229845523834229, |
|
-0.4998578727245331, |
|
-0.8867913484573364, |
|
1.4196479014572105e-06 |
|
], |
|
"q01": [ |
|
-0.01119564864784479, |
|
-0.014266146533191203, |
|
-0.0071747214533388615, |
|
-0.009444301575422287, |
|
-0.03990109823644161, |
|
-0.017422311007976532, |
|
4.003279136668425e-05 |
|
], |
|
"q99": [ |
|
0.01015154086053368, |
|
0.017181577533483497, |
|
0.007216989761218411, |
|
0.010380979906767595, |
|
0.03556173853576176, |
|
0.018032474815845446, |
|
0.9982578039169312 |
|
], |
|
"std": [ |
|
0.04264938458800316, |
|
0.04428559169173241, |
|
0.12224084138870239, |
|
0.005388413090258837, |
|
0.011246449314057827, |
|
0.006287882570177317, |
|
0.39732322096824646 |
|
] |
|
}, |
|
"num_trajectories": 5208, |
|
"num_transitions": 1139911, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"fmb_dataset": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
1.399999976158142, |
|
1.0, |
|
1.399999976158142, |
|
1.0, |
|
1.0, |
|
1.0, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.059029702097177505, |
|
-0.06476633995771408, |
|
-0.09787475317716599, |
|
0.004325388930737972, |
|
0.00028963794466108084, |
|
-0.04457257315516472, |
|
0.7336440086364746 |
|
], |
|
"min": [ |
|
-1.399999976158142, |
|
-1.399999976158142, |
|
-1.0, |
|
-1.0, |
|
-1.0, |
|
-1.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.8257142901420593, |
|
-1.399999976158142, |
|
-1.0, |
|
-1.0, |
|
-0.3028571307659149, |
|
-1.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
1.0, |
|
0.5257142782211304, |
|
1.0, |
|
1.0, |
|
0.3400000035762787, |
|
1.0, |
|
1.0 |
|
], |
|
"std": [ |
|
0.28809213638305664, |
|
0.2820415794849396, |
|
0.4626740515232086, |
|
0.3266514539718628, |
|
0.10842999070882797, |
|
0.3440099358558655, |
|
0.4435282051563263 |
|
] |
|
}, |
|
"num_trajectories": 8612, |
|
"num_transitions": 1137459, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"fractal20220817_data": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
2.9984593391418457, |
|
22.09052848815918, |
|
2.7507524490356445, |
|
1.570636510848999, |
|
1.5321086645126343, |
|
1.5691522359848022, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.006987582892179489, |
|
0.006265917327255011, |
|
-0.01262515690177679, |
|
0.04333311319351196, |
|
-0.005756212864071131, |
|
0.0009130256366916001, |
|
0.5354204773902893 |
|
], |
|
"min": [ |
|
-2.0204520225524902, |
|
-5.497899532318115, |
|
-2.031663417816162, |
|
-1.569917917251587, |
|
-1.569892168045044, |
|
-1.570419430732727, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.22453527510166169, |
|
-0.14820013284683228, |
|
-0.231589707583189, |
|
-0.3517994859814644, |
|
-0.4193011274933815, |
|
-0.43643461108207704, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.17824687153100965, |
|
0.14938379630446405, |
|
0.21842354819178575, |
|
0.5892666035890578, |
|
0.35272657424211445, |
|
0.44796681255102094, |
|
1.0 |
|
], |
|
"std": [ |
|
0.0692116990685463, |
|
0.05970962345600128, |
|
0.07353084534406662, |
|
0.15610496699810028, |
|
0.13164450228214264, |
|
0.14593800902366638, |
|
0.497110515832901 |
|
] |
|
}, |
|
"num_trajectories": 87212, |
|
"num_transitions": 3786400, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"furniture_bench_dataset_converted_externally_to_rlds": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.10000000149011612, |
|
0.10000000149011612, |
|
0.10000000149011612, |
|
0.8651833534240723, |
|
1.0909736156463623, |
|
2.863185405731201, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.00014610752987209707, |
|
0.0010830952087417245, |
|
0.0006224989192560315, |
|
-0.003303206292912364, |
|
-0.0026880695950239897, |
|
0.018242603167891502, |
|
0.48854944109916687 |
|
], |
|
"min": [ |
|
-0.10495579987764359, |
|
-0.10939455777406693, |
|
-0.10000000149011612, |
|
-0.971906840801239, |
|
-1.0475432872772217, |
|
-3.06000018119812, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.053988199681043625, |
|
-0.05049169331789017, |
|
-0.032499241530895236, |
|
-0.1953887003660202, |
|
-0.41674559473991396, |
|
-0.8886768388748169, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.05414841488003723, |
|
0.04965164884924884, |
|
0.060055799782276154, |
|
0.18231668293476103, |
|
0.39867786407470646, |
|
0.8772023963928218, |
|
1.0 |
|
], |
|
"std": [ |
|
0.01610708422958851, |
|
0.014891477301716805, |
|
0.014014219865202904, |
|
0.058274295181035995, |
|
0.11417088657617569, |
|
0.33479776978492737, |
|
0.49991825222969055 |
|
] |
|
}, |
|
"num_trajectories": 5100, |
|
"num_transitions": 3948057, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"iamlab_cmu_pickup_insert_converted_externally_to_rlds": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.6634981632232666, |
|
0.23428471386432648, |
|
0.4308285415172577, |
|
3.1415927410125732, |
|
0.13647015392780304, |
|
3.141592502593994, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.5274372696876526, |
|
0.02858201041817665, |
|
0.18712575733661652, |
|
1.2339589595794678, |
|
0.03226623684167862, |
|
-1.4199490547180176, |
|
0.5550631880760193 |
|
], |
|
"min": [ |
|
0.3071657121181488, |
|
-0.29754969477653503, |
|
0.06578229367733002, |
|
-3.1415927410125732, |
|
-0.04584203287959099, |
|
-3.141592502593994, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.3148897051811218, |
|
-0.20317550599575043, |
|
0.06785467118024827, |
|
-3.140952730178833, |
|
-0.029743434861302376, |
|
-3.141091251373291, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.6472805738449097, |
|
0.20846802592277527, |
|
0.36855655312538155, |
|
3.1409926891326903, |
|
0.11424950212240226, |
|
3.1410969257354737, |
|
1.0 |
|
], |
|
"std": [ |
|
0.08108345419168472, |
|
0.1116757020354271, |
|
0.07747554779052734, |
|
2.8737246990203857, |
|
0.02774704433977604, |
|
2.7678682804107666, |
|
0.49695101380348206 |
|
] |
|
}, |
|
"num_trajectories": 631, |
|
"num_transitions": 146241, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"jaco_play": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.20000000298023224, |
|
0.20000000298023224, |
|
0.20000000298023224, |
|
0.0, |
|
0.0, |
|
0.0, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.0009658430935814977, |
|
-0.00580078037455678, |
|
-0.00395062193274498, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.34934908151626587 |
|
], |
|
"min": [ |
|
-0.20000000298023224, |
|
-0.20000000298023224, |
|
-0.20000000298023224, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.20000000298023224, |
|
-0.20000000298023224, |
|
-0.20000000298023224, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.20000000298023224, |
|
0.20000000298023224, |
|
0.20000000298023224, |
|
0.0, |
|
0.0, |
|
0.0, |
|
1.0 |
|
], |
|
"std": [ |
|
0.12235074490308762, |
|
0.09678777307271957, |
|
0.11155334860086441, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.4768252968788147 |
|
] |
|
}, |
|
"num_trajectories": 1085, |
|
"num_transitions": 77965, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"kuka": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.1697135865688324, |
|
0.2777623236179352, |
|
0.43710532784461975, |
|
0.0, |
|
0.0, |
|
1.9684287309646606, |
|
1.0 |
|
], |
|
"mean": [ |
|
-0.0004668905457947403, |
|
0.00040138536132872105, |
|
-0.001280792523175478, |
|
0.0, |
|
0.0, |
|
-0.03722453489899635, |
|
0.4131543040275574 |
|
], |
|
"min": [ |
|
-0.159867063164711, |
|
-0.2892282009124756, |
|
-0.2795473635196686, |
|
0.0, |
|
0.0, |
|
-1.9875637292861938, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.06619441494345665, |
|
-0.08713878810405731, |
|
-0.15083016991615295, |
|
0.0, |
|
0.0, |
|
-0.5415697038173676, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.06601839080452929, |
|
0.08732476785779003, |
|
0.18168179214000715, |
|
0.0, |
|
0.0, |
|
0.2923380345106127, |
|
1.0 |
|
], |
|
"std": [ |
|
0.02083250693976879, |
|
0.02915887162089348, |
|
0.06422865390777588, |
|
0.0, |
|
0.0, |
|
0.14224295318126678, |
|
0.49086448550224304 |
|
] |
|
}, |
|
"num_trajectories": 209880, |
|
"num_transitions": 2455879, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"nyu_franka_play_dataset_converted_externally_to_rlds": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.06424188613891602, |
|
0.07027634978294373, |
|
0.06129661202430725, |
|
6.281067848205566, |
|
0.1967729926109314, |
|
0.26377415657043457, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.001021989737637341, |
|
-0.00012002651783404872, |
|
0.00032894269679673016, |
|
0.0015034361276775599, |
|
-0.002198522910475731, |
|
-0.001663230243138969, |
|
0.7230083346366882 |
|
], |
|
"min": [ |
|
-0.05952230095863342, |
|
-0.07232445478439331, |
|
-0.06730806827545166, |
|
-6.278434753417969, |
|
-0.21479034423828125, |
|
-0.3627619743347168, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.03199600875377655, |
|
-0.032861671447753905, |
|
-0.03368805110454559, |
|
-0.12080862045288086, |
|
-0.12175218224525451, |
|
-0.11370223641395569, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.03101520001888276, |
|
0.0373908892273903, |
|
0.03646374464035038, |
|
0.11764093399047852, |
|
0.1258920183777809, |
|
0.09366151213645942, |
|
1.0 |
|
], |
|
"std": [ |
|
0.01327415369451046, |
|
0.013215910643339157, |
|
0.012822109274566174, |
|
0.2732451558113098, |
|
0.057022541761398315, |
|
0.039172880351543427, |
|
0.44752755761146545 |
|
] |
|
}, |
|
"num_trajectories": 456, |
|
"num_transitions": 44875, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"roboturk": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.39124172925949097, |
|
0.4601028263568878, |
|
0.4870833456516266, |
|
1.816888689994812, |
|
1.8240282535552979, |
|
1.4824820756912231, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.0014448732836171985, |
|
-0.0015945249469950795, |
|
-0.0011753785656765103, |
|
0.0023012510500848293, |
|
-0.0009382463176734746, |
|
-0.00011485807772260159, |
|
0.5746025443077087 |
|
], |
|
"min": [ |
|
-0.6546999216079712, |
|
-0.6365841031074524, |
|
-0.4217723608016968, |
|
-1.6695482730865479, |
|
-1.8023357391357422, |
|
-1.4630827903747559, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.1342635464668274, |
|
-0.19996687173843383, |
|
-0.1482972100377083, |
|
-0.20720748245716095, |
|
-0.09676413893699647, |
|
-0.18075634717941286, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.14956976801157001, |
|
0.1805950567126275, |
|
0.18841815620660796, |
|
0.21615413755178453, |
|
0.09457383215427405, |
|
0.18543301910162005, |
|
1.0 |
|
], |
|
"std": [ |
|
0.04935386776924133, |
|
0.0635455846786499, |
|
0.061164740473032, |
|
0.09553450345993042, |
|
0.08420111238956451, |
|
0.06517903506755829, |
|
0.49452081322669983 |
|
] |
|
}, |
|
"num_trajectories": 1995, |
|
"num_transitions": 187507, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"stanford_hydra_dataset_converted_externally_to_rlds": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.02499854564666748, |
|
0.02499903365969658, |
|
0.024999922141432762, |
|
0.24974457919597626, |
|
0.24997030198574066, |
|
0.24999946355819702, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.0007790001109242439, |
|
0.00013707754260394722, |
|
-0.0002548607881180942, |
|
0.0012903271708637476, |
|
-0.004751681815832853, |
|
0.002692886395379901, |
|
0.48855218291282654 |
|
], |
|
"min": [ |
|
-0.024999044835567474, |
|
-0.024999700486660004, |
|
-0.02499929815530777, |
|
-0.24993225932121277, |
|
-0.2499666064977646, |
|
-0.2499932497739792, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.019992006458342076, |
|
-0.02415412735193968, |
|
-0.022941758055239916, |
|
-0.11085530579090118, |
|
-0.12024572037160397, |
|
-0.13314770206809043, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.022886231057345868, |
|
0.022358838934451335, |
|
0.02410089675337076, |
|
0.12370114490389822, |
|
0.11323311634361738, |
|
0.18474749639630164, |
|
1.0 |
|
], |
|
"std": [ |
|
0.008022161200642586, |
|
0.009131459519267082, |
|
0.009574338793754578, |
|
0.04122216999530792, |
|
0.0384303517639637, |
|
0.04606688767671585, |
|
0.49976691603660583 |
|
] |
|
}, |
|
"num_trajectories": 570, |
|
"num_transitions": 358234, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"taco_play": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
1.4915844202041626, |
|
2.1842432022094727, |
|
2.6836395263671875, |
|
5.035226821899414, |
|
2.665864944458008, |
|
4.250768661499023, |
|
1.0 |
|
], |
|
"mean": [ |
|
-0.003845922416076064, |
|
0.009671456180512905, |
|
0.012780580669641495, |
|
-0.005403771996498108, |
|
-0.009606587700545788, |
|
-0.002480733208358288, |
|
0.4263913035392761 |
|
], |
|
"min": [ |
|
-4.242457866668701, |
|
-3.192805051803589, |
|
-1.3371467590332031, |
|
-4.202683448791504, |
|
-2.6722638607025146, |
|
-3.3467135429382324, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.7106140398979186, |
|
-1.056944659948349, |
|
-0.5878450274467468, |
|
-0.7682853937149048, |
|
-0.7180147767066956, |
|
-1.5527938604354858, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.6482916426658629, |
|
1.0051310062408447, |
|
0.9480248689651489, |
|
0.6926478147506714, |
|
0.6351067513227462, |
|
1.628010264635086, |
|
1.0 |
|
], |
|
"std": [ |
|
0.23254038393497467, |
|
0.36298269033432007, |
|
0.28692901134490967, |
|
0.2617705166339874, |
|
0.2438892275094986, |
|
0.5216503143310547, |
|
0.4946896731853485 |
|
] |
|
}, |
|
"num_trajectories": 3603, |
|
"num_transitions": 237798, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"toto": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
0.6839867234230042, |
|
0.4454185664653778, |
|
0.7984078526496887, |
|
2.120781660079956, |
|
1.371164321899414, |
|
1.4118704795837402, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.38542115688323975, |
|
0.007769413758069277, |
|
0.3632740378379822, |
|
-0.6652036905288696, |
|
0.1890396922826767, |
|
0.03298724442720413, |
|
0.0 |
|
], |
|
"min": [ |
|
0.09922284632921219, |
|
-0.5180193781852722, |
|
0.13791072368621826, |
|
-2.635117530822754, |
|
-1.0734480619430542, |
|
-1.9282547235488892, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.1756722891330719, |
|
-0.3077590811252594, |
|
0.235383919775486, |
|
-2.0908505964279174, |
|
-0.6191593289375306, |
|
-0.7488683319091797, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.6136963081359863, |
|
0.33704194784164443, |
|
0.6681221985816956, |
|
0.7422861719131538, |
|
0.7955395007133507, |
|
0.740464625358582, |
|
0.0 |
|
], |
|
"std": [ |
|
0.12211652100086212, |
|
0.19378550350666046, |
|
0.10178236663341522, |
|
0.5725259184837341, |
|
0.29884573817253113, |
|
0.3259911835193634, |
|
0.0 |
|
] |
|
}, |
|
"num_trajectories": 1003, |
|
"num_transitions": 325699, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"ucsd_kitchen_dataset_converted_externally_to_rlds": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
678.0, |
|
400.0, |
|
507.0, |
|
180.00001525878906, |
|
6.000013828277588, |
|
116.99998474121094, |
|
1.0 |
|
], |
|
"mean": [ |
|
410.37567138671875, |
|
116.9518814086914, |
|
192.35032653808594, |
|
-121.22441864013672, |
|
-33.84893035888672, |
|
50.016136169433594, |
|
0.741813600063324 |
|
], |
|
"min": [ |
|
172.0, |
|
-166.0, |
|
-99.99999237060547, |
|
-180.00001525878906, |
|
-89.0, |
|
-96.00010681152344, |
|
0.0 |
|
], |
|
"q01": [ |
|
200.00001052856445, |
|
-102.31004211425781, |
|
-94.99993370056153, |
|
-180.00001525878906, |
|
-88.00001525878906, |
|
-38.999977111816406, |
|
0.0 |
|
], |
|
"q99": [ |
|
637.0, |
|
368.30999999999995, |
|
493.0, |
|
180.00001525878906, |
|
0.999983012676239, |
|
105.00001525878906, |
|
1.0 |
|
], |
|
"std": [ |
|
122.81494903564453, |
|
108.8009033203125, |
|
130.303466796875, |
|
116.28205108642578, |
|
27.621843338012695, |
|
41.02094650268555, |
|
0.43763357400894165 |
|
] |
|
}, |
|
"num_trajectories": 150, |
|
"num_transitions": 3970, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"utaustin_mutex": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
1.0, |
|
1.0, |
|
1.0, |
|
0.375, |
|
0.375, |
|
0.375, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.06176406890153885, |
|
-0.005005486309528351, |
|
0.10216785222291946, |
|
-0.03314131125807762, |
|
0.013895004987716675, |
|
-0.011317633092403412, |
|
0.5038976669311523 |
|
], |
|
"min": [ |
|
-1.0, |
|
-1.0, |
|
-1.0, |
|
-0.375, |
|
-0.375, |
|
-0.375, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.4285714328289032, |
|
-0.9800000190734863, |
|
-0.5571428537368774, |
|
-0.375, |
|
-0.15642857551574707, |
|
-0.335357129573822, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.5914285778999329, |
|
0.9714285731315613, |
|
1.0, |
|
0.3278571367263794, |
|
0.207857146859169, |
|
0.25607141852378845, |
|
1.0 |
|
], |
|
"std": [ |
|
0.1875014752149582, |
|
0.4468473494052887, |
|
0.3792876601219177, |
|
0.14097853004932404, |
|
0.06453701853752136, |
|
0.11765272170305252, |
|
0.501045286655426 |
|
] |
|
}, |
|
"num_trajectories": 1500, |
|
"num_transitions": 361883, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
}, |
|
"viola": { |
|
"action": { |
|
"mask": [ |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
true, |
|
false |
|
], |
|
"max": [ |
|
1.0, |
|
1.0, |
|
1.0, |
|
0.375, |
|
0.36321428418159485, |
|
0.375, |
|
1.0 |
|
], |
|
"mean": [ |
|
0.04761844128370285, |
|
-0.029204415157437325, |
|
0.05586736649274826, |
|
-0.002618510741740465, |
|
0.006867344491183758, |
|
-0.01682133786380291, |
|
0.7323777675628662 |
|
], |
|
"min": [ |
|
-1.0, |
|
-1.0, |
|
-1.0, |
|
-0.375, |
|
-0.375, |
|
-0.375, |
|
0.0 |
|
], |
|
"q01": [ |
|
-0.9628571271896362, |
|
-1.0, |
|
-1.0, |
|
-0.26249998807907104, |
|
-0.21321429312229156, |
|
-0.3385714292526245, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.9114285707473755, |
|
0.868571400642395, |
|
1.0, |
|
0.2817857265472412, |
|
0.2239285707473755, |
|
0.3557142913341522, |
|
1.0 |
|
], |
|
"std": [ |
|
0.39157867431640625, |
|
0.4076525568962097, |
|
0.40077948570251465, |
|
0.10023996233940125, |
|
0.0844319611787796, |
|
0.10375042259693146, |
|
0.44260647892951965 |
|
] |
|
}, |
|
"num_trajectories": 150, |
|
"num_transitions": 76324, |
|
"proprio": { |
|
"max": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"mean": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"min": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q01": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"q99": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
], |
|
"std": [ |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0, |
|
0.0 |
|
] |
|
} |
|
} |
|
}, |
|
"output_projector_states": false, |
|
"pad_to_multiple_of": 64, |
|
"pad_token_id": 32000, |
|
"text_config": { |
|
"model_type": "llama", |
|
"pad_token_id": 32000, |
|
"torch_dtype": "bfloat16", |
|
"vocab_size": 32064 |
|
}, |
|
"timm_model_ids": [ |
|
"vit_large_patch14_reg4_dinov2.lvd142m", |
|
"vit_so400m_patch14_siglip_224" |
|
], |
|
"timm_override_act_layers": [ |
|
null, |
|
null |
|
], |
|
"torch_dtype": "bfloat16", |
|
"transformers_version": "4.40.1", |
|
"use_fused_vision_backbone": true, |
|
"vision_backbone_id": "dinosiglip-vit-so-224px" |
|
} |
|
|