jat-project/jat-dataset-tokenized

Name: jat-project/jat-dataset-tokenized
Creator: jat-project
Published: 2023-12-22 22:17:42
License: 暂无描述

Hugging Face2023-12-22 更新2024-06-22 收录

下载链接：

https://hf-mirror.com/datasets/jat-project/jat-dataset-tokenized

下载链接

链接失效反馈

官方服务：

资源简介：

--- dataset_info: - config_name: atari-alien features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51686398456 num_examples: 14134 - name: test num_bytes: 5412188320 num_examples: 1480 download_size: 847071867 dataset_size: 57098586776 - config_name: atari-amidar features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 52362921996 num_examples: 14319 - name: test num_bytes: 4808802460 num_examples: 1315 download_size: 645217608 dataset_size: 57171724456 - config_name: atari-assault features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 52757865468 num_examples: 14427 - name: test num_bytes: 4421172756 num_examples: 1209 download_size: 253415283 dataset_size: 57179038224 - config_name: atari-asterix features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 52863915104 num_examples: 14456 - name: test num_bytes: 5137922020 num_examples: 1405 download_size: 293282697 dataset_size: 58001837124 - config_name: atari-asteroids features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 52468971632 num_examples: 14348 - name: test num_bytes: 3605687624 num_examples: 986 download_size: 316908651 dataset_size: 56074659256 - config_name: atari-atlantis features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 52384863300 num_examples: 14325 - name: test num_bytes: 3975032908 num_examples: 1087 download_size: 274032418 dataset_size: 56359896208 - config_name: atari-bankheist features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51807075628 num_examples: 14167 - name: test num_bytes: 5836386864 num_examples: 1596 download_size: 879900687 dataset_size: 57643462492 - config_name: atari-battlezone features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51126895204 num_examples: 13981 - name: test num_bytes: 6092368744 num_examples: 1666 download_size: 530266996 dataset_size: 57219263948 - config_name: atari-beamrider features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 49155834728 num_examples: 13442 - name: test num_bytes: 7880585020 num_examples: 2155 download_size: 427025312 dataset_size: 57036419748 - config_name: atari-berzerk features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 49492268056 num_examples: 13534 - name: test num_bytes: 6172820192 num_examples: 1688 download_size: 351445377 dataset_size: 55665088248 - config_name: atari-bowling features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51598633240 num_examples: 14110 - name: test num_bytes: 5898553892 num_examples: 1613 download_size: 163624131 dataset_size: 57497187132 - config_name: atari-boxing features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 53178407128 num_examples: 14542 - name: test num_bytes: 5883926356 num_examples: 1609 download_size: 662704435 dataset_size: 59062333484 - config_name: atari-breakout features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 49272855016 num_examples: 13474 - name: test num_bytes: 6611646272 num_examples: 1808 download_size: 265049647 dataset_size: 55884501288 - config_name: atari-centipede features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51913125264 num_examples: 14196 - name: test num_bytes: 6026544832 num_examples: 1648 download_size: 269104472 dataset_size: 57939670096 - config_name: atari-choppercommand features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 48991274948 num_examples: 13397 - name: test num_bytes: 7156521988 num_examples: 1957 download_size: 425086559 dataset_size: 56147796936 - config_name: atari-crazyclimber features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51291454984 num_examples: 14026 - name: test num_bytes: 5712052808 num_examples: 1562 download_size: 458314909 dataset_size: 57003507792 - config_name: atari-defender features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 49382561536 num_examples: 13504 - name: test num_bytes: 6172820192 num_examples: 1688 download_size: 217534779 dataset_size: 55555381728 - config_name: atari-demonattack features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 49364277116 num_examples: 13499 - name: test num_bytes: 6172820192 num_examples: 1688 download_size: 209141226 dataset_size: 55537097308 - config_name: atari-doubledunk features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 5799818024 num_examples: 1586 - name: train num_bytes: 52264186128 num_examples: 14292 download_size: 585265286 dataset_size: 58064004152 - config_name: atari-enduro features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 48490281840 num_examples: 13260 - name: test num_bytes: 6172820192 num_examples: 1688 download_size: 696314069 dataset_size: 54663102032 - config_name: atari-fishingderby features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51463328532 num_examples: 14073 - name: test num_bytes: 6085054976 num_examples: 1664 download_size: 817608846 dataset_size: 57548383508 - config_name: atari-freeway features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51254886144 num_examples: 14016 - name: test num_bytes: 5851014400 num_examples: 1600 download_size: 684669809 dataset_size: 57105900544 - config_name: atari-frostbite features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51470642300 num_examples: 14075 - name: test num_bytes: 5898553892 num_examples: 1613 download_size: 629892834 dataset_size: 57369196192 - config_name: atari-gopher features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 48062426412 num_examples: 13143 - name: test num_bytes: 6436115840 num_examples: 1760 download_size: 278315347 dataset_size: 54498542252 - config_name: atari-gravitar features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 52677414020 num_examples: 14405 - name: test num_bytes: 5927808964 num_examples: 1621 download_size: 297931288 dataset_size: 58605222984 - config_name: atari-hero features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51357278896 num_examples: 14044 - name: test num_bytes: 5891240124 num_examples: 1611 download_size: 467961084 dataset_size: 57248519020 - config_name: atari-icehockey features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51258543028 num_examples: 14017 - name: test num_bytes: 5876612588 num_examples: 1607 download_size: 369055326 dataset_size: 57135155616 - config_name: atari-jamesbond features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 46361975352 num_examples: 12678 - name: test num_bytes: 10352638604 num_examples: 2831 download_size: 485679287 dataset_size: 56714613956 - config_name: atari-kangaroo features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 52103283232 num_examples: 14248 - name: test num_bytes: 5638915128 num_examples: 1542 download_size: 427266047 dataset_size: 57742198360 - config_name: atari-krull features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51942380336 num_examples: 14204 - name: test num_bytes: 5807131792 num_examples: 1588 download_size: 1439632028 dataset_size: 57749512128 - config_name: atari-kungfumaster features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51306082520 num_examples: 14030 - name: test num_bytes: 6136251352 num_examples: 1678 download_size: 689596673 dataset_size: 57442333872 - config_name: atari-montezumarevenge features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51997233596 num_examples: 14219 - name: test num_bytes: 5924152080 num_examples: 1620 download_size: 739361910 dataset_size: 57921385676 - config_name: atari-mspacman features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51635202080 num_examples: 14120 - name: test num_bytes: 5664513316 num_examples: 1549 download_size: 867194250 dataset_size: 57299715396 - config_name: atari-namethisgame features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 49642200300 num_examples: 13575 - name: test num_bytes: 6874941920 num_examples: 1880 download_size: 520921217 dataset_size: 56517142220 - config_name: atari-phoenix features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 49510552476 num_examples: 13539 - name: test num_bytes: 6172820192 num_examples: 1688 download_size: 241965818 dataset_size: 55683372668 - config_name: atari-pitfall features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 52245901708 num_examples: 14287 - name: test num_bytes: 4812459344 num_examples: 1316 download_size: 385040106 dataset_size: 57058361052 - config_name: atari-pong features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 5894897008 num_examples: 1612 - name: train num_bytes: 51748565484 num_examples: 14151 download_size: 128206463 dataset_size: 57643462492 - config_name: atari-privateeye features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 5902210776 num_examples: 1614 - name: train num_bytes: 51580348820 num_examples: 14105 download_size: 762572093 dataset_size: 57482559596 - config_name: atari-qbert features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 5715709692 num_examples: 1563 - name: train num_bytes: 51291454984 num_examples: 14026 download_size: 697728392 dataset_size: 57007164676 - config_name: atari-riverraid features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 5437786508 num_examples: 1487 - name: train num_bytes: 52202019100 num_examples: 14275 download_size: 685859297 dataset_size: 57639805608 - config_name: atari-roadrunner features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 5774219836 num_examples: 1579 - name: train num_bytes: 51660800268 num_examples: 14127 download_size: 463497648 dataset_size: 57435020104 - config_name: atari-robotank features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 5090382528 num_examples: 1392 - name: train num_bytes: 51485269836 num_examples: 14079 download_size: 471559799 dataset_size: 56575652364 - config_name: atari-seaquest features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 5730337228 num_examples: 1567 - name: train num_bytes: 51551093748 num_examples: 14097 download_size: 328551402 dataset_size: 57281430976 - config_name: atari-skiing features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 53785449872 num_examples: 14708 - name: test num_bytes: 6000946644 num_examples: 1641 download_size: 567502031 dataset_size: 59786396516 - config_name: atari-solaris features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51924095916 num_examples: 14199 - name: test num_bytes: 5233001004 num_examples: 1431 download_size: 492333967 dataset_size: 57157096920 - config_name: atari-spaceinvaders features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 46266896368 num_examples: 12652 - name: test num_bytes: 9548124124 num_examples: 2611 download_size: 300389865 dataset_size: 55815020492 - config_name: atari-stargunner features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 50545450648 num_examples: 13822 - name: test num_bytes: 5865641936 num_examples: 1604 download_size: 203075318 dataset_size: 56411092584 - config_name: atari-surround features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 50611274560 num_examples: 13840 - name: test num_bytes: 6381262580 num_examples: 1745 download_size: 286861481 dataset_size: 56992537140 - config_name: atari-tennis features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 51423102808 num_examples: 14062 - name: test num_bytes: 5675483968 num_examples: 1552 download_size: 407941157 dataset_size: 57098586776 - config_name: atari-timepilot features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 50816060064 num_examples: 13896 - name: test num_bytes: 5759592300 num_examples: 1575 download_size: 285156447 dataset_size: 56575652364 - config_name: atari-tutankham features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 47981974964 num_examples: 13121 - name: test num_bytes: 8140223784 num_examples: 2226 download_size: 382912419 dataset_size: 56122198748 - config_name: atari-upndown features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 49382561536 num_examples: 13504 - name: test num_bytes: 6172820192 num_examples: 1688 download_size: 1690613769 dataset_size: 55555381728 - config_name: atari-venture features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 5313452452 num_examples: 1453 - name: train num_bytes: 52147165840 num_examples: 14260 download_size: 509488474 dataset_size: 57460618292 - config_name: atari-videopinball features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 1996658664 num_examples: 546 - name: train num_bytes: 52191048448 num_examples: 14272 download_size: 605138140 dataset_size: 54187707112 - config_name: atari-wizardofwor features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 6033858600 num_examples: 1650 - name: train num_bytes: 50903825280 num_examples: 13920 download_size: 646859311 dataset_size: 56937683880 - config_name: atari-yarsrevenge features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 6000946644 num_examples: 1641 - name: train num_bytes: 51126895204 num_examples: 13981 download_size: 1424379144 dataset_size: 57127841848 - config_name: atari-zaxxon features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 6088711860 num_examples: 1665 - name: train num_bytes: 50585676372 num_examples: 13833 download_size: 452125956 dataset_size: 56674388232 - config_name: babyai-action-obj-door features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 128870282 dataset_size: 43957200000 - config_name: babyai-blocked-unlock-pickup features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 137033255 dataset_size: 43957200000 - config_name: babyai-boss-level features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2236102764 num_examples: 5087 - name: train num_bytes: 42505293684 num_examples: 96697 download_size: 344912338 dataset_size: 44741396448 - config_name: babyai-boss-level-no-unlock features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2217640740 num_examples: 5045 - name: train num_bytes: 42103964448 num_examples: 95784 download_size: 339304020 dataset_size: 44321605188 - config_name: babyai-find-obj-s5 features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 133212544 dataset_size: 43957200000 - config_name: babyai-go-to features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 233927543 dataset_size: 43957200000 - config_name: babyai-go-to-door features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 118992586 dataset_size: 43957200000 - config_name: babyai-go-to-imp-unlock features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 43664005476 num_examples: 99333 - name: test num_bytes: 891012444 num_examples: 2027 download_size: 366460821 dataset_size: 44555017920 - config_name: babyai-go-to-local features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 130476854 dataset_size: 43957200000 - config_name: babyai-go-to-obj features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 122037932 dataset_size: 43957200000 - config_name: babyai-go-to-obj-door features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 133904822 dataset_size: 43957200000 - config_name: babyai-go-to-red-ball features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 107941553 dataset_size: 43957200000 - config_name: babyai-go-to-red-ball-grey features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 108701381 dataset_size: 43957200000 - config_name: babyai-go-to-red-ball-no-dists features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 100751341 dataset_size: 43957200000 - config_name: babyai-go-to-red-blue-ball features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41759340000 num_examples: 95000 - name: test num_bytes: 2197860000 num_examples: 5000 download_size: 109835377 dataset_size: 43957200000 - config_name: babyai-go-to-seq features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 41792307900 num_examples: 95075 - name: test num_bytes: 2198739144 num_examples: 5002 download_size: 288118166 dataset_size: 43991047044 - config_name: babyai-key-corridor features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 273451937 dataset_size: 43957200000 - config_name: babyai-mini-boss-level features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2200497432 num_examples: 5006 - name: train num_bytes: 41821759224 num_examples: 95142 download_size: 167867886 dataset_size: 44022256656 - config_name: babyai-move-two-across-s8n9 features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 268471454 dataset_size: 43957200000 - config_name: babyai-one-room-s8 features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 101603110 dataset_size: 43957200000 - config_name: babyai-open features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 181194361 dataset_size: 43957200000 - config_name: babyai-open-door features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 127824190 dataset_size: 43957200000 - config_name: babyai-open-doors-order-n4 features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 127418529 dataset_size: 43957200000 - config_name: babyai-open-red-door features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 78248393 dataset_size: 43957200000 - config_name: babyai-open-two-doors features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 130542191 dataset_size: 43957200000 - config_name: babyai-pickup features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 236053290 dataset_size: 43957200000 - config_name: babyai-pickup-above features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 163058824 dataset_size: 43957200000 - config_name: babyai-pickup-dist features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2077856844 num_examples: 4727 - name: train num_bytes: 39403234080 num_examples: 89640 download_size: 114895484 dataset_size: 41481090924 - config_name: babyai-pickup-loc features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 134221714 dataset_size: 43957200000 - config_name: babyai-put-next features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 43078056000 num_examples: 98000 - name: test num_bytes: 879144000 num_examples: 2000 download_size: 169889411 dataset_size: 43957200000 - config_name: babyai-put-next-local features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 43078056000 num_examples: 98000 - name: test num_bytes: 879144000 num_examples: 2000 download_size: 157089711 dataset_size: 43957200000 - config_name: babyai-synth features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41765054436 num_examples: 95013 download_size: 231769022 dataset_size: 43962914436 - config_name: babyai-synth-loc features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2198739144 num_examples: 5002 - name: train num_bytes: 41766373152 num_examples: 95016 download_size: 245211619 dataset_size: 43965112296 - config_name: babyai-synth-seq features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2207530584 num_examples: 5022 - name: train num_bytes: 41981763432 num_examples: 95506 download_size: 326087180 dataset_size: 44189294016 - config_name: babyai-unblock-pickup features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41765933580 num_examples: 95015 download_size: 241680488 dataset_size: 43963793580 - config_name: babyai-unlock features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 43259159664 num_examples: 98412 - name: test num_bytes: 883979292 num_examples: 2011 download_size: 328757743 dataset_size: 44143138956 - config_name: babyai-unlock-local features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 116723486 dataset_size: 43957200000 - config_name: babyai-unlock-pickup features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 2197860000 num_examples: 5000 - name: train num_bytes: 41759340000 num_examples: 95000 download_size: 137214787 dataset_size: 43957200000 - config_name: babyai-unlock-to-unlock features: - name: discrete_observations sequence: sequence: int64 - name: discrete_actions sequence: int64 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 43078056000 num_examples: 98000 - name: test num_bytes: 879144000 num_examples: 2000 download_size: 158735389 dataset_size: 43957200000 - config_name: conceptual-captions features: - name: input_ids sequence: int32 - name: attention_mask sequence: int8 - name: pixel_values sequence: sequence: sequence: float32 - name: loss_weight sequence: float32 splits: - name: test num_bytes: 7574631480 num_examples: 12465 - name: train num_bytes: 303836000000 num_examples: 500000 download_size: 82071298648 dataset_size: 311410631480 - config_name: metaworld-assembly features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 64267084 dataset_size: 851910400 - config_name: metaworld-basketball features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 162412290 dataset_size: 851910400 - config_name: metaworld-bin-picking features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 168127631 dataset_size: 851910400 - config_name: metaworld-box-close features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 174656572 dataset_size: 851910400 - config_name: metaworld-button-press features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 106951062 dataset_size: 851910400 - config_name: metaworld-button-press-topdown features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 117078197 dataset_size: 851910400 - config_name: metaworld-button-press-topdown-wall features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 119641275 dataset_size: 851910400 - config_name: metaworld-button-press-wall features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 112458551 dataset_size: 851910400 - config_name: metaworld-coffee-button features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 112608052 dataset_size: 851910400 - config_name: metaworld-coffee-pull features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 161591807 dataset_size: 851910400 - config_name: metaworld-coffee-push features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 173247466 dataset_size: 851910400 - config_name: metaworld-dial-turn features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 102519630 dataset_size: 851910400 - config_name: metaworld-disassemble features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 72920062 dataset_size: 851910400 - config_name: metaworld-door-close features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 153530521 dataset_size: 851910400 - config_name: metaworld-door-lock features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 123855874 dataset_size: 851910400 - config_name: metaworld-door-open features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 140905068 dataset_size: 851910400 - config_name: metaworld-door-unlock features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 121700706 dataset_size: 851910400 - config_name: metaworld-drawer-close features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 101417660 dataset_size: 851910400 - config_name: metaworld-drawer-open features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 96573298 dataset_size: 851910400 - config_name: metaworld-faucet-close features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 89353472 dataset_size: 851910400 - config_name: metaworld-faucet-open features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 96651789 dataset_size: 851910400 - config_name: metaworld-hammer features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 177539984 dataset_size: 851910400 - config_name: metaworld-hand-insert features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 135665012 dataset_size: 851910400 - config_name: metaworld-handle-press features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 103407785 dataset_size: 851910400 - config_name: metaworld-handle-press-side features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 103403469 dataset_size: 851910400 - config_name: metaworld-handle-pull features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 121440284 dataset_size: 851910400 - config_name: metaworld-handle-pull-side features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 118413651 dataset_size: 851910400 - config_name: metaworld-lever-pull features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 168776851 dataset_size: 851910400 - config_name: metaworld-peg-insert-side features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 153705593 dataset_size: 851910400 - config_name: metaworld-peg-unplug-side features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 171742157 dataset_size: 851910400 - config_name: metaworld-pick-out-of-hole features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 22274303 dataset_size: 851910400 - config_name: metaworld-pick-place features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 176678495 dataset_size: 851910400 - config_name: metaworld-pick-place-wall features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 172257534 dataset_size: 851910400 - config_name: metaworld-plate-slide features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 114432287 dataset_size: 851910400 - config_name: metaworld-plate-slide-back features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 36662627 dataset_size: 851910400 - config_name: metaworld-plate-slide-back-side features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 33762161 dataset_size: 851910400 - config_name: metaworld-plate-slide-side features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 106392923 dataset_size: 851910400 - config_name: metaworld-push features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 166180034 dataset_size: 851910400 - config_name: metaworld-push-back features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 133027374 dataset_size: 851910400 - config_name: metaworld-push-wall features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 158267234 dataset_size: 851910400 - config_name: metaworld-reach features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 168663459 dataset_size: 851910400 - config_name: metaworld-reach-wall features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 171608203 dataset_size: 851910400 - config_name: metaworld-shelf-place features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 142334952 dataset_size: 851910400 - config_name: metaworld-soccer features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 159081606 dataset_size: 851910400 - config_name: metaworld-stick-pull features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 170289154 dataset_size: 851910400 - config_name: metaworld-stick-push features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 166125948 dataset_size: 851910400 - config_name: metaworld-sweep features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 164632354 dataset_size: 851910400 - config_name: metaworld-sweep-into features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 135177252 dataset_size: 851910400 - config_name: metaworld-window-close features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 95044772 dataset_size: 851910400 - config_name: metaworld-window-open features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 774464000 num_examples: 16000 - name: test num_bytes: 77446400 num_examples: 1600 download_size: 95793720 dataset_size: 851910400 - config_name: mujoco-ant features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 1420167204 num_examples: 35317 - name: test num_bytes: 158435280 num_examples: 3940 download_size: 1513512326 dataset_size: 1578602484 - config_name: mujoco-doublependulum features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 599126920 num_examples: 35962 - name: test num_bytes: 66490060 num_examples: 3991 download_size: 458306888 dataset_size: 665616980 - config_name: mujoco-halfcheetah features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 1005264000 num_examples: 36000 - name: test num_bytes: 111696000 num_examples: 4000 download_size: 1055030042 dataset_size: 1116960000 - config_name: mujoco-hopper features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 377714520 num_examples: 20190 - name: test num_bytes: 41774964 num_examples: 2233 download_size: 343653363 dataset_size: 419489484 - config_name: mujoco-humanoid features: - name: continuous_observations sequence: sequence: float32 - name: rewards sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 13565692988 num_examples: 33347 - name: test num_bytes: 1509649644 num_examples: 3711 download_size: 10439047554 dataset_size: 15075342632 - config_name: mujoco-pendulum features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 201391764 num_examples: 21217 - name: test num_bytes: 22334676 num_examples: 2353 download_size: 134650231 dataset_size: 223726440 - config_name: mujoco-pusher features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 315828000 num_examples: 9000 - name: test num_bytes: 35092000 num_examples: 1000 download_size: 134738418 dataset_size: 350920000 - config_name: mujoco-reacher features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 159156000 num_examples: 9000 - name: test num_bytes: 17684000 num_examples: 1000 download_size: 38441946 dataset_size: 176840000 - config_name: mujoco-standup features: - name: rewards sequence: float32 - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 14644944000 num_examples: 36000 - name: test num_bytes: 1627216000 num_examples: 4000 download_size: 11711102671 dataset_size: 16272160000 - config_name: mujoco-swimmer features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 526032000 num_examples: 36000 - name: test num_bytes: 58448000 num_examples: 4000 download_size: 519559720 dataset_size: 584480000 - config_name: mujoco-walker features: - name: continuous_observations sequence: sequence: float32 - name: continuous_actions sequence: sequence: float32 - name: rewards sequence: float32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 944529300 num_examples: 33825 - name: test num_bytes: 104798772 num_examples: 3753 download_size: 954326371 dataset_size: 1049328072 - config_name: ok-vqa features: - name: input_ids sequence: int32 - name: attention_mask sequence: int8 - name: pixel_values sequence: sequence: sequence: float32 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 5474517048 num_examples: 9009 - name: test num_bytes: 3066312912 num_examples: 5046 download_size: 2461083826 dataset_size: 8540829960 - config_name: oscar features: - name: input_ids sequence: int32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 58269773100 num_examples: 12612505 - name: test num_bytes: 63899220 num_examples: 13831 download_size: 10788173669 dataset_size: 58333672320 - config_name: wikipedia features: - name: input_ids sequence: int32 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: train num_bytes: 59293939320 num_examples: 12834186 - name: test num_bytes: 58216620 num_examples: 12601 download_size: 10100547139 dataset_size: 59352155940 configs: - config_name: atari-alien data_files: - split: train path: atari-alien/train-* - split: test path: atari-alien/test-* - config_name: atari-amidar data_files: - split: train path: atari-amidar/train-* - split: test path: atari-amidar/test-* - config_name: atari-assault data_files: - split: train path: atari-assault/train-* - split: test path: atari-assault/test-* - config_name: atari-asterix data_files: - split: train path: atari-asterix/train-* - split: test path: atari-asterix/test-* - config_name: atari-asteroids data_files: - split: train path: atari-asteroids/train-* - split: test path: atari-asteroids/test-* - config_name: atari-atlantis data_files: - split: train path: atari-atlantis/train-* - split: test path: atari-atlantis/test-* - config_name: atari-bankheist data_files: - split: train path: atari-bankheist/train-* - split: test path: atari-bankheist/test-* - config_name: atari-battlezone data_files: - split: train path: atari-battlezone/train-* - split: test path: atari-battlezone/test-* - config_name: atari-beamrider data_files: - split: train path: atari-beamrider/train-* - split: test path: atari-beamrider/test-* - config_name: atari-berzerk data_files: - split: train path: atari-berzerk/train-* - split: test path: atari-berzerk/test-* - config_name: atari-bowling data_files: - split: train path: atari-bowling/train-* - split: test path: atari-bowling/test-* - config_name: atari-boxing data_files: - split: train path: atari-boxing/train-* - split: test path: atari-boxing/test-* - config_name: atari-breakout data_files: - split: train path: atari-breakout/train-* - split: test path: atari-breakout/test-* - config_name: atari-centipede data_files: - split: train path: atari-centipede/train-* - split: test path: atari-centipede/test-* - config_name: atari-choppercommand data_files: - split: train path: atari-choppercommand/train-* - split: test path: atari-choppercommand/test-* - config_name: atari-crazyclimber data_files: - split: train path: atari-crazyclimber/train-* - split: test path: atari-crazyclimber/test-* - config_name: atari-defender data_files: - split: train path: atari-defender/train-* - split: test path: atari-defender/test-* - config_name: atari-demonattack data_files: - split: train path: atari-demonattack/train-* - split: test path: atari-demonattack/test-* - config_name: atari-doubledunk data_files: - split: test path: atari-doubledunk/test-* - split: train path: atari-doubledunk/train-* - config_name: atari-enduro data_files: - split: train path: atari-enduro/train-* - split: test path: atari-enduro/test-* - config_name: atari-fishingderby data_files: - split: train path: atari-fishingderby/train-* - split: test path: atari-fishingderby/test-* - config_name: atari-freeway data_files: - split: train path: atari-freeway/train-* - split: test path: atari-freeway/test-* - config_name: atari-frostbite data_files: - split: train path: atari-frostbite/train-* - split: test path: atari-frostbite/test-* - config_name: atari-gopher data_files: - split: train path: atari-gopher/train-* - split: test path: atari-gopher/test-* - config_name: atari-gravitar data_files: - split: train path: atari-gravitar/train-* - split: test path: atari-gravitar/test-* - config_name: atari-hero data_files: - split: train path: atari-hero/train-* - split: test path: atari-hero/test-* - config_name: atari-icehockey data_files: - split: train path: atari-icehockey/train-* - split: test path: atari-icehockey/test-* - config_name: atari-jamesbond data_files: - split: train path: atari-jamesbond/train-* - split: test path: atari-jamesbond/test-* - config_name: atari-kangaroo data_files: - split: train path: atari-kangaroo/train-* - split: test path: atari-kangaroo/test-* - config_name: atari-krull data_files: - split: train path: atari-krull/train-* - split: test path: atari-krull/test-* - config_name: atari-kungfumaster data_files: - split: train path: atari-kungfumaster/train-* - split: test path: atari-kungfumaster/test-* - config_name: atari-montezumarevenge data_files: - split: train path: atari-montezumarevenge/train-* - split: test path: atari-montezumarevenge/test-* - config_name: atari-mspacman data_files: - split: train path: atari-mspacman/train-* - split: test path: atari-mspacman/test-* - config_name: atari-namethisgame data_files: - split: train path: atari-namethisgame/train-* - split: test path: atari-namethisgame/test-* - config_name: atari-phoenix data_files: - split: train path: atari-phoenix/train-* - split: test path: atari-phoenix/test-* - config_name: atari-pitfall data_files: - split: train path: atari-pitfall/train-* - split: test path: atari-pitfall/test-* - config_name: atari-pong data_files: - split: test path: atari-pong/test-* - split: train path: atari-pong/train-* - config_name: atari-privateeye data_files: - split: test path: atari-privateeye/test-* - split: train path: atari-privateeye/train-* - config_name: atari-qbert data_files: - split: test path: atari-qbert/test-* - split: train path: atari-qbert/train-* - config_name: atari-riverraid data_files: - split: test path: atari-riverraid/test-* - split: train path: atari-riverraid/train-* - config_name: atari-roadrunner data_files: - split: test path: atari-roadrunner/test-* - split: train path: atari-roadrunner/train-* - config_name: atari-robotank data_files: - split: test path: atari-robotank/test-* - split: train path: atari-robotank/train-* - config_name: atari-seaquest data_files: - split: test path: atari-seaquest/test-* - split: train path: atari-seaquest/train-* - config_name: atari-skiing data_files: - split: train path: atari-skiing/train-* - split: test path: atari-skiing/test-* - config_name: atari-solaris data_files: - split: train path: atari-solaris/train-* - split: test path: atari-solaris/test-* - config_name: atari-spaceinvaders data_files: - split: train path: atari-spaceinvaders/train-* - split: test path: atari-spaceinvaders/test-* - config_name: atari-stargunner data_files: - split: train path: atari-stargunner/train-* - split: test path: atari-stargunner/test-* - config_name: atari-surround data_files: - split: train path: atari-surround/train-* - split: test path: atari-surround/test-* - config_name: atari-tennis data_files: - split: train path: atari-tennis/train-* - split: test path: atari-tennis/test-* - config_name: atari-timepilot data_files: - split: train path: atari-timepilot/train-* - split: test path: atari-timepilot/test-* - config_name: atari-tutankham data_files: - split: train path: atari-tutankham/train-* - split: test path: atari-tutankham/test-* - config_name: atari-upndown data_files: - split: train path: atari-upndown/train-* - split: test path: atari-upndown/test-* - config_name: atari-venture data_files: - split: test path: atari-venture/test-* - split: train path: atari-venture/train-* - config_name: atari-videopinball data_files: - split: test path: atari-videopinball/test-* - split: train path: atari-videopinball/train-* - config_name: atari-wizardofwor data_files: - split: test path: atari-wizardofwor/test-* - split: train path: atari-wizardofwor/train-* - config_name: atari-yarsrevenge data_files: - split: test path: atari-yarsrevenge/test-* - split: train path: atari-yarsrevenge/train-* - config_name: atari-zaxxon data_files: - split: test path: atari-zaxxon/test-* - split: train path: atari-zaxxon/train-* - config_name: babyai-action-obj-door data_files: - split: train path: babyai-action-obj-door/train-* - split: test path: babyai-action-obj-door/test-* - config_name: babyai-blocked-unlock-pickup data_files: - split: test path: babyai-blocked-unlock-pickup/test-* - split: train path: babyai-blocked-unlock-pickup/train-* - config_name: babyai-boss-level data_files: - split: test path: babyai-boss-level/test-* - split: train path: babyai-boss-level/train-* - config_name: babyai-boss-level-no-unlock data_files: - split: test path: babyai-boss-level-no-unlock/test-* - split: train path: babyai-boss-level-no-unlock/train-* - config_name: babyai-find-obj-s5 data_files: - split: train path: babyai-find-obj-s5/train-* - split: test path: babyai-find-obj-s5/test-* - config_name: babyai-go-to data_files: - split: train path: babyai-go-to/train-* - split: test path: babyai-go-to/test-* - config_name: babyai-go-to-door data_files: - split: train path: babyai-go-to-door/train-* - split: test path: babyai-go-to-door/test-* - config_name: babyai-go-to-imp-unlock data_files: - split: train path: babyai-go-to-imp-unlock/train-* - split: test path: babyai-go-to-imp-unlock/test-* - config_name: babyai-go-to-local data_files: - split: train path: babyai-go-to-local/train-* - split: test path: babyai-go-to-local/test-* - config_name: babyai-go-to-obj data_files: - split: train path: babyai-go-to-obj/train-* - split: test path: babyai-go-to-obj/test-* - config_name: babyai-go-to-obj-door data_files: - split: train path: babyai-go-to-obj-door/train-* - split: test path: babyai-go-to-obj-door/test-* - config_name: babyai-go-to-red-ball data_files: - split: train path: babyai-go-to-red-ball/train-* - split: test path: babyai-go-to-red-ball/test-* - config_name: babyai-go-to-red-ball-grey data_files: - split: train path: babyai-go-to-red-ball-grey/train-* - split: test path: babyai-go-to-red-ball-grey/test-* - config_name: babyai-go-to-red-ball-no-dists data_files: - split: train path: babyai-go-to-red-ball-no-dists/train-* - split: test path: babyai-go-to-red-ball-no-dists/test-* - config_name: babyai-go-to-red-blue-ball data_files: - split: train path: babyai-go-to-red-blue-ball/train-* - split: test path: babyai-go-to-red-blue-ball/test-* - config_name: babyai-go-to-seq data_files: - split: train path: babyai-go-to-seq/train-* - split: test path: babyai-go-to-seq/test-* - config_name: babyai-key-corridor data_files: - split: test path: babyai-key-corridor/test-* - split: train path: babyai-key-corridor/train-* - config_name: babyai-mini-boss-level data_files: - split: test path: babyai-mini-boss-level/test-* - split: train path: babyai-mini-boss-level/train-* - config_name: babyai-move-two-across-s8n9 data_files: - split: test path: babyai-move-two-across-s8n9/test-* - split: train path: babyai-move-two-across-s8n9/train-* - config_name: babyai-one-room-s8 data_files: - split: test path: babyai-one-room-s8/test-* - split: train path: babyai-one-room-s8/train-* - config_name: babyai-open data_files: - split: test path: babyai-open/test-* - split: train path: babyai-open/train-* - config_name: babyai-open-door data_files: - split: test path: babyai-open-door/test-* - split: train path: babyai-open-door/train-* - config_name: babyai-open-doors-order-n4 data_files: - split: test path: babyai-open-doors-order-n4/test-* - split: train path: babyai-open-doors-order-n4/train-* - config_name: babyai-open-red-door data_files: - split: test path: babyai-open-red-door/test-* - split: train path: babyai-open-red-door/train-* - config_name: babyai-open-two-doors data_files: - split: test path: babyai-open-two-doors/test-* - split: train path: babyai-open-two-doors/train-* - config_name: babyai-pickup data_files: - split: test path: babyai-pickup/test-* - split: train path: babyai-pickup/train-* - config_name: babyai-pickup-above data_files: - split: test path: babyai-pickup-above/test-* - split: train path: babyai-pickup-above/train-* - config_name: babyai-pickup-dist data_files: - split: test path: babyai-pickup-dist/test-* - split: train path: babyai-pickup-dist/train-* - config_name: babyai-pickup-loc data_files: - split: test path: babyai-pickup-loc/test-* - split: train path: babyai-pickup-loc/train-* - config_name: babyai-put-next data_files: - split: train path: babyai-put-next/train-* - split: test path: babyai-put-next/test-* - config_name: babyai-put-next-local data_files: - split: train path: babyai-put-next-local/train-* - split: test path: babyai-put-next-local/test-* - config_name: babyai-synth data_files: - split: test path: babyai-synth/test-* - split: train path: babyai-synth/train-* - config_name: babyai-synth-loc data_files: - split: test path: babyai-synth-loc/test-* - split: train path: babyai-synth-loc/train-* - config_name: babyai-synth-seq data_files: - split: test path: babyai-synth-seq/test-* - split: train path: babyai-synth-seq/train-* - config_name: babyai-unblock-pickup data_files: - split: test path: babyai-unblock-pickup/test-* - split: train path: babyai-unblock-pickup/train-* - config_name: babyai-unlock data_files: - split: train path: babyai-unlock/train-* - split: test path: babyai-unlock/test-* - config_name: babyai-unlock-local data_files: - split: test path: babyai-unlock-local/test-* - split: train path: babyai-unlock-local/train-* - config_name: babyai-unlock-pickup data_files: - split: test path: babyai-unlock-pickup/test-* - split: train path: babyai-unlock-pickup/train-* - config_name: babyai-unlock-to-unlock data_files: - split: train path: babyai-unlock-to-unlock/train-* - split: test path: babyai-unlock-to-unlock/test-* - config_name: conceptual-captions data_files: - split: test path: conceptual-captions/test-* - split: train path: conceptual-captions/train-* - config_name: metaworld-assembly data_files: - split: train path: metaworld-assembly/train-* - split: test path: metaworld-assembly/test-* - config_name: metaworld-basketball data_files: - split: train path: metaworld-basketball/train-* - split: test path: metaworld-basketball/test-* - config_name: metaworld-bin-picking data_files: - split: train path: metaworld-bin-picking/train-* - split: test path: metaworld-bin-picking/test-* - config_name: metaworld-box-close data_files: - split: train path: metaworld-box-close/train-* - split: test path: metaworld-box-close/test-* - config_name: metaworld-button-press data_files: - split: train path: metaworld-button-press/train-* - split: test path: metaworld-button-press/test-* - config_name: metaworld-button-press-topdown data_files: - split: train path: metaworld-button-press-topdown/train-* - split: test path: metaworld-button-press-topdown/test-* - config_name: metaworld-button-press-topdown-wall data_files: - split: train path: metaworld-button-press-topdown-wall/train-* - split: test path: metaworld-button-press-topdown-wall/test-* - config_name: metaworld-button-press-wall data_files: - split: train path: metaworld-button-press-wall/train-* - split: test path: metaworld-button-press-wall/test-* - config_name: metaworld-coffee-button data_files: - split: train path: metaworld-coffee-button/train-* - split: test path: metaworld-coffee-button/test-* - config_name: metaworld-coffee-pull data_files: - split: train path: metaworld-coffee-pull/train-* - split: test path: metaworld-coffee-pull/test-* - config_name: metaworld-coffee-push data_files: - split: train path: metaworld-coffee-push/train-* - split: test path: metaworld-coffee-push/test-* - config_name: metaworld-dial-turn data_files: - split: train path: metaworld-dial-turn/train-* - split: test path: metaworld-dial-turn/test-* - config_name: metaworld-disassemble data_files: - split: train path: metaworld-disassemble/train-* - split: test path: metaworld-disassemble/test-* - config_name: metaworld-door-close data_files: - split: train path: metaworld-door-close/train-* - split: test path: metaworld-door-close/test-* - config_name: metaworld-door-lock data_files: - split: train path: metaworld-door-lock/train-* - split: test path: metaworld-door-lock/test-* - config_name: metaworld-door-open data_files: - split: train path: metaworld-door-open/train-* - split: test path: metaworld-door-open/test-* - config_name: metaworld-door-unlock data_files: - split: train path: metaworld-door-unlock/train-* - split: test path: metaworld-door-unlock/test-* - config_name: metaworld-drawer-close data_files: - split: train path: metaworld-drawer-close/train-* - split: test path: metaworld-drawer-close/test-* - config_name: metaworld-drawer-open data_files: - split: train path: metaworld-drawer-open/train-* - split: test path: metaworld-drawer-open/test-* - config_name: metaworld-faucet-close data_files: - split: train path: metaworld-faucet-close/train-* - split: test path: metaworld-faucet-close/test-* - config_name: metaworld-faucet-open data_files: - split: train path: metaworld-faucet-open/train-* - split: test path: metaworld-faucet-open/test-* - config_name: metaworld-hammer data_files: - split: train path: metaworld-hammer/train-* - split: test path: metaworld-hammer/test-* - config_name: metaworld-hand-insert data_files: - split: train path: metaworld-hand-insert/train-* - split: test path: metaworld-hand-insert/test-* - config_name: metaworld-handle-press data_files: - split: train path: metaworld-handle-press/train-* - split: test path: metaworld-handle-press/test-* - config_name: metaworld-handle-press-side data_files: - split: train path: metaworld-handle-press-side/train-* - split: test path: metaworld-handle-press-side/test-* - config_name: metaworld-handle-pull data_files: - split: train path: metaworld-handle-pull/train-* - split: test path: metaworld-handle-pull/test-* - config_name: metaworld-handle-pull-side data_files: - split: train path: metaworld-handle-pull-side/train-* - split: test path: metaworld-handle-pull-side/test-* - config_name: metaworld-lever-pull data_files: - split: train path: metaworld-lever-pull/train-* - split: test path: metaworld-lever-pull/test-* - config_name: metaworld-peg-insert-side data_files: - split: train path: metaworld-peg-insert-side/train-* - split: test path: metaworld-peg-insert-side/test-* - config_name: metaworld-peg-unplug-side data_files: - split: train path: metaworld-peg-unplug-side/train-* - split: test path: metaworld-peg-unplug-side/test-* - config_name: metaworld-pick-out-of-hole data_files: - split: train path: metaworld-pick-out-of-hole/train-* - split: test path: metaworld-pick-out-of-hole/test-* - config_name: metaworld-pick-place data_files: - split: train path: metaworld-pick-place/train-* - split: test path: metaworld-pick-place/test-* - config_name: metaworld-pick-place-wall data_files: - split: train path: metaworld-pick-place-wall/train-* - split: test path: metaworld-pick-place-wall/test-* - config_name: metaworld-plate-slide data_files: - split: train path: metaworld-plate-slide/train-* - split: test path: metaworld-plate-slide/test-* - config_name: metaworld-plate-slide-back data_files: - split: train path: metaworld-plate-slide-back/train-* - split: test path: metaworld-plate-slide-back/test-* - config_name: metaworld-plate-slide-back-side data_files: - split: train path: metaworld-plate-slide-back-side/train-* - split: test path: metaworld-plate-slide-back-side/test-* - config_name: metaworld-plate-slide-side data_files: - split: train path: metaworld-plate-slide-side/train-* - split: test path: metaworld-plate-slide-side/test-* - config_name: metaworld-push data_files: - split: train path: metaworld-push/train-* - split: test path: metaworld-push/test-* - config_name: metaworld-push-back data_files: - split: train path: metaworld-push-back/train-* - split: test path: metaworld-push-back/test-* - config_name: metaworld-push-wall data_files: - split: train path: metaworld-push-wall/train-* - split: test path: metaworld-push-wall/test-* - config_name: metaworld-reach data_files: - split: train path: metaworld-reach/train-* - split: test path: metaworld-reach/test-* - config_name: metaworld-reach-wall data_files: - split: train path: metaworld-reach-wall/train-* - split: test path: metaworld-reach-wall/test-* - config_name: metaworld-shelf-place data_files: - split: train path: metaworld-shelf-place/train-* - split: test path: metaworld-shelf-place/test-* - config_name: metaworld-soccer data_files: - split: train path: metaworld-soccer/train-* - split: test path: metaworld-soccer/test-* - config_name: metaworld-stick-pull data_files: - split: train path: metaworld-stick-pull/train-* - split: test path: metaworld-stick-pull/test-* - config_name: metaworld-stick-push data_files: - split: train path: metaworld-stick-push/train-* - split: test path: metaworld-stick-push/test-* - config_name: metaworld-sweep data_files: - split: train path: metaworld-sweep/train-* - split: test path: metaworld-sweep/test-* - config_name: metaworld-sweep-into data_files: - split: train path: metaworld-sweep-into/train-* - split: test path: metaworld-sweep-into/test-* - config_name: metaworld-window-close data_files: - split: train path: metaworld-window-close/train-* - split: test path: metaworld-window-close/test-* - config_name: metaworld-window-open data_files: - split: train path: metaworld-window-open/train-* - split: test path: metaworld-window-open/test-* - config_name: mujoco-ant data_files: - split: train path: mujoco-ant/train-* - split: test path: mujoco-ant/test-* - config_name: mujoco-doublependulum data_files: - split: train path: mujoco-doublependulum/train-* - split: test path: mujoco-doublependulum/test-* - config_name: mujoco-halfcheetah data_files: - split: train path: mujoco-halfcheetah/train-* - split: test path: mujoco-halfcheetah/test-* - config_name: mujoco-hopper data_files: - split: train path: mujoco-hopper/train-* - split: test path: mujoco-hopper/test-* - config_name: mujoco-humanoid data_files: - split: train path: mujoco-humanoid/train-* - split: test path: mujoco-humanoid/test-* - config_name: mujoco-pendulum data_files: - split: train path: mujoco-pendulum/train-* - split: test path: mujoco-pendulum/test-* - config_name: mujoco-pusher data_files: - split: train path: mujoco-pusher/train-* - split: test path: mujoco-pusher/test-* - config_name: mujoco-reacher data_files: - split: train path: mujoco-reacher/train-* - split: test path: mujoco-reacher/test-* - config_name: mujoco-standup data_files: - split: train path: mujoco-standup/train-* - split: test path: mujoco-standup/test-* - config_name: mujoco-swimmer data_files: - split: train path: mujoco-swimmer/train-* - split: test path: mujoco-swimmer/test-* - config_name: mujoco-walker data_files: - split: train path: mujoco-walker/train-* - split: test path: mujoco-walker/test-* - config_name: ok-vqa data_files: - split: train path: ok-vqa/train-* - split: test path: ok-vqa/test-* - config_name: oscar data_files: - split: train path: oscar/train-* - split: test path: oscar/test-* - config_name: wikipedia data_files: - split: train path: wikipedia/train-* - split: test path: wikipedia/test-* --- # Dataset Card for "jat-dataset-tokenized" [More Information needed](https://github.com/huggingface/datasets/blob/main/CONTRIBUTING.md#how-to-contribute-to-the-dataset-cards)

提供机构：

jat-project

原始信息汇总

数据集概述

该数据集包含多个配置，每个配置对应不同的Atari游戏。以下是各配置的详细信息：

配置列表

atari-alien
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为51686398456，样本数为14134
  - test: 字节数为5412188320，样本数为1480
- 下载大小：847071867字节
- 数据集大小：57098586776字节
atari-amidar
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为52362921996，样本数为14319
  - test: 字节数为4808802460，样本数为1315
- 下载大小：645217608字节
- 数据集大小：57171724456字节
atari-assault
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为52757865468，样本数为14427
  - test: 字节数为4421172756，样本数为1209
- 下载大小：253415283字节
- 数据集大小：57179038224字节
atari-asterix
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为52863915104，样本数为14456
  - test: 字节数为5137922020，样本数为1405
- 下载大小：293282697字节
- 数据集大小：58001837124字节
atari-asteroids
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为52468971632，样本数为14348
  - test: 字节数为3605687624，样本数为986
- 下载大小：316908651字节
- 数据集大小：56074659256字节
atari-atlantis
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为52384863300，样本数为14325
  - test: 字节数为3975032908，样本数为1087
- 下载大小：274032418字节
- 数据集大小：56359896208字节
atari-bankheist
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为51807075628，样本数为14167
  - test: 字节数为5836386864，样本数为1596
- 下载大小：879900687字节
- 数据集大小：57643462492字节
atari-battlezone
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为51126895204，样本数为13981
  - test: 字节数为6092368744，样本数为1666
- 下载大小：530266996字节
- 数据集大小：57219263948字节
atari-beamrider
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为49155834728，样本数为13442
  - test: 字节数为7880585020，样本数为2155
- 下载大小：427025312字节
- 数据集大小：57036419748字节
atari-berzerk
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为49492268056，样本数为13534
  - test: 字节数为6172820192，样本数为1688
- 下载大小：351445377字节
- 数据集大小：55665088248字节
atari-bowling
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为51598633240，样本数为14110
  - test: 字节数为5898553892，样本数为1613
- 下载大小：163624131字节
- 数据集大小：57497187132字节
atari-boxing
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为53178407128，样本数为14542
  - test: 字节数为5883926356，样本数为1609
- 下载大小：662704435字节
- 数据集大小：59062333484字节
atari-breakout
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为49272855016，样本数为13474
  - test: 字节数为6611646272，样本数为1808
- 下载大小：265049647字节
- 数据集大小：55884501288字节
atari-centipede
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为51913125264，样本数为14196
  - test: 字节数为6026544832，样本数为1648
- 下载大小：269104472字节
- 数据集大小：57939670096字节
atari-choppercommand
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为48991274948，样本数为13397
  - test: 字节数为7156521988，样本数为1957
- 下载大小：425086559字节
- 数据集大小：56147796936字节
atari-crazyclimber
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为51291454984，样本数为14026
  - test: 字节数为5712052808，样本数为1562
- 下载大小：458314909字节
- 数据集大小：57003507792字节
atari-defender
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为49382561536，样本数为13504
  - test: 字节数为6172820192，样本数为1688
- 下载大小：217534779字节
- 数据集大小：55555381728字节
atari-demonattack
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为49364277116，样本数为13499
  - test: 字节数为6172820192，样本数为1688
- 下载大小：209141226字节
- 数据集大小：55537097308字节
atari-doubledunk
- 特征：
  - image_observations: 多层序列，最终类型为float32
  - rewards: 序列，类型为float32
  - discrete_actions: 序列，类型为int64
  - attention_mask: 序列，类型为int8
  - loss_weight: 序列，类型为float32
- 分割：
  - train: 字节数为52264186128，样本数为14292
  - test: 字节数为5799818024，样本数为1586
- 下载大小：585265286字节
- 数据集大小：58064004152字节
atari-enduro

搜集汇总

数据集介绍

背景与挑战

背景概述

jat-dataset-tokenized是一个时间序列数据集，格式为parquet，包含多个子集，适用于机器学习和AI研究，特别是游戏和模拟环境相关任务。

以上内容由遇见数据集搜集并总结生成

5,000+

优质数据集

54 个

任务类型

进入经典数据集