region:us
---
dataset_info:
- config_name: atari-alien
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22936842328
    num_examples: 17354
  - name: test
    num_bytes: 2346887392
    num_examples: 1776
  download_size: 2054563645
  dataset_size: 25283729720
- config_name: atari-amidar
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23517570100
    num_examples: 17811
  - name: test
    num_bytes: 2149432108
    num_examples: 1629
  download_size: 1633498377
  dataset_size: 25667002208
- config_name: atari-assault
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22970100628
    num_examples: 17355
  - name: test
    num_bytes: 1940979184
    num_examples: 1468
  download_size: 758593716
  dataset_size: 24911079812
- config_name: atari-asterix
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 24875418768
    num_examples: 18996
  - name: test
    num_bytes: 2505413272
    num_examples: 1914
  download_size: 1028844936
  dataset_size: 27380832040
- config_name: atari-asteroids
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22628281880
    num_examples: 17074
  - name: test
    num_bytes: 1551124052
    num_examples: 1171
  download_size: 860484458
  dataset_size: 24179405932
- config_name: atari-atlantis
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22717495892
    num_examples: 17155
  - name: test
    num_bytes: 1720200176
    num_examples: 1300
  download_size: 797937432
  dataset_size: 24437696068
- config_name: atari-bankheist
  features:
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23503705888
    num_examples: 17872
  - name: test
    num_bytes: 2542380684
    num_examples: 1933
  download_size: 2805364640
  dataset_size: 26046086572
- config_name: atari-battlezone
  features:
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22340903992
    num_examples: 16898
  - name: test
    num_bytes: 2679685428
    num_examples: 2027
  download_size: 1520563037
  dataset_size: 25020589420
- config_name: atari-beamrider
  features:
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21331851988
    num_examples: 16115
  - name: test
    num_bytes: 3443729560
    num_examples: 2602
  download_size: 1408532322
  dataset_size: 24775581548
- config_name: atari-berzerk
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21397562160
    num_examples: 16148
  - name: test
    num_bytes: 2655628912
    num_examples: 2004
  download_size: 877619752
  dataset_size: 24053191072
- config_name: atari-bowling
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23312520876
    num_examples: 17677
  - name: test
    num_bytes: 2609218872
    num_examples: 1978
  download_size: 562079900
  dataset_size: 25921739748
- config_name: atari-boxing
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 26396397960
    num_examples: 20334
  - name: test
    num_bytes: 3095765472
    num_examples: 2376
  download_size: 1556486472
  dataset_size: 29492163432
- config_name: atari-breakout
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21288271588
    num_examples: 16071
  - name: test
    num_bytes: 2881835492
    num_examples: 2175
  download_size: 738709382
  dataset_size: 24170107080
- config_name: atari-centipede
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 24756206272
    num_examples: 18904
  - name: test
    num_bytes: 2808894852
    num_examples: 2143
  download_size: 896456615
  dataset_size: 27565101124
- config_name: atari-choppercommand
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21815683640
    num_examples: 16522
  - name: test
    num_bytes: 3196036720
    num_examples: 2420
  download_size: 1118565629
  dataset_size: 25011720360
- config_name: atari-crazyclimber
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22616477812
    num_examples: 17099
  - name: test
    num_bytes: 2573496856
    num_examples: 1946
  download_size: 1500494388
  dataset_size: 25189974668
- config_name: atari-defender
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21281584216
    num_examples: 16058
  - name: test
    num_bytes: 2656206064
    num_examples: 2004
  download_size: 683940397
  dataset_size: 23937790280
- config_name: atari-demonattack
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21317852508
    num_examples: 16089
  - name: test
    num_bytes: 2690768548
    num_examples: 2031
  download_size: 649507529
  dataset_size: 24008621056
- config_name: atari-doubledunk
  features:
  - name: local_positions
    sequence: int64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 25267613728
    num_examples: 19312
  - name: test
    num_bytes: 2712595536
    num_examples: 2076
  download_size: 1480348122
  dataset_size: 27980209264
- config_name: atari-enduro
  features:
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 20958634024
    num_examples: 15814
  - name: test
    num_bytes: 2655965840
    num_examples: 2004
  download_size: 1496020338
  dataset_size: 23614599864
- config_name: atari-fishingderby
  features:
  - name: local_positions
    sequence: int64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 24047962676
    num_examples: 18283
  - name: test
    num_bytes: 2889068116
    num_examples: 2195
  download_size: 2219957113
  dataset_size: 26937030792
- config_name: atari-freeway
  features:
  - name: local_positions
    sequence: int64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23371276840
    num_examples: 17678
  - name: test
    num_bytes: 2556328396
    num_examples: 1933
  download_size: 1792035914
  dataset_size: 25927605236
- config_name: atari-frostbite
  features:
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22904514944
    num_examples: 17368
  - name: test
    num_bytes: 2669353472
    num_examples: 2024
  download_size: 1644837128
  dataset_size: 25573868416
- config_name: atari-gopher
  features:
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 20793543720
    num_examples: 15694
  - name: test
    num_bytes: 2768439836
    num_examples: 2089
  download_size: 814329245
  dataset_size: 23561983556
- config_name: atari-gravitar
  features:
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 25847429280
    num_examples: 19872
  - name: test
    num_bytes: 2828307480
    num_examples: 2178
  download_size: 922784336
  dataset_size: 28675736760
- config_name: atari-hero
  features:
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22800954176
    num_examples: 17280
  - name: test
    num_bytes: 2566446396
    num_examples: 1945
  download_size: 1210979007
  dataset_size: 25367400572
- config_name: atari-icehockey
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22674337348
    num_examples: 17159
  - name: test
    num_bytes: 2610457248
    num_examples: 1976
  download_size: 888965775
  dataset_size: 25284794596
- config_name: atari-jamesbond
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 20352364304
    num_examples: 15380
  - name: test
    num_bytes: 4463759572
    num_examples: 3371
  download_size: 1224656422
  dataset_size: 24816123876
- config_name: atari-kangaroo
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 25096596968
    num_examples: 19166
  - name: test
    num_bytes: 2650226840
    num_examples: 2026
  download_size: 1297855371
  dataset_size: 27746823808
- config_name: atari-krull
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 24220658504
    num_examples: 18422
  - name: test
    num_bytes: 2618995228
    num_examples: 1993
  download_size: 3531737628
  dataset_size: 26839653732
- config_name: atari-kungfumaster
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22995045044
    num_examples: 17419
  - name: test
    num_bytes: 2701969412
    num_examples: 2047
  download_size: 2577334191
  dataset_size: 25697014456
- config_name: atari-montezumarevenge
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 24412433468
    num_examples: 18601
  - name: test
    num_bytes: 2735262644
    num_examples: 2083
  download_size: 2702911146
  dataset_size: 27147696112
- config_name: atari-mspacman
  features:
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23368802440
    num_examples: 17718
  - name: test
    num_bytes: 2534137148
    num_examples: 1921
  download_size: 2274887896
  dataset_size: 25902939588
- config_name: atari-namethisgame
  features:
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21644294212
    num_examples: 16351
  - name: test
    num_bytes: 2989141496
    num_examples: 2258
  download_size: 1298483978
  dataset_size: 24633435708
- config_name: atari-phoenix
  features:
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21388349940
    num_examples: 16139
  - name: test
    num_bytes: 2655773200
    num_examples: 2004
  download_size: 716057486
  dataset_size: 24044123140
- config_name: atari-pitfall
  features:
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22794223464
    num_examples: 17214
  - name: test
    num_bytes: 2148675844
    num_examples: 1623
  download_size: 1019075298
  dataset_size: 24942899308
- config_name: atari-pong
  features:
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23550084136
    num_examples: 17934
  - name: test
    num_bytes: 2749678664
    num_examples: 2094
  download_size: 511973690
  dataset_size: 26299762800
- config_name: atari-privateeye
  features:
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23352851484
    num_examples: 17665
  - name: test
    num_bytes: 2547222824
    num_examples: 1926
  download_size: 1765155315
  dataset_size: 25900074308
- config_name: atari-qbert
  features:
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22492555216
    num_examples: 17020
  - name: test
    num_bytes: 2515704280
    num_examples: 1906
  download_size: 1731005653
  dataset_size: 25008259496
- config_name: atari-riverraid
  features:
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23991409520
    num_examples: 18228
  - name: test
    num_bytes: 2507541488
    num_examples: 1908
  download_size: 1814647945
  dataset_size: 26498951008
- config_name: atari-roadrunner
  features:
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23312301308
    num_examples: 17689
  - name: test
    num_bytes: 2660211924
    num_examples: 2019
  download_size: 1291914824
  dataset_size: 25972513232
- config_name: atari-robotank
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22742071628
    num_examples: 17189
  - name: test
    num_bytes: 2228137996
    num_examples: 1685
  download_size: 1315268257
  dataset_size: 24970209624
- config_name: atari-seaquest
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23269421504
    num_examples: 17680
  - name: test
    num_bytes: 2545247816
    num_examples: 1934
  download_size: 884373274
  dataset_size: 25814669320
- config_name: atari-skiing
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 26768165764
    num_examples: 20711
  - name: test
    num_bytes: 2982150188
    num_examples: 2309
  download_size: 1828629595
  dataset_size: 29750315952
- config_name: atari-solaris
  features:
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22578149564
    num_examples: 17049
  - name: test
    num_bytes: 2292090424
    num_examples: 1730
  download_size: 1280172003
  dataset_size: 24870239988
- config_name: atari-spaceinvaders
  features:
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 20199993304
    num_examples: 15250
  - name: test
    num_bytes: 4108004592
    num_examples: 3100
  download_size: 858101809
  dataset_size: 24307997896
- config_name: atari-stargunner
  features:
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22090952068
    num_examples: 16679
  - name: test
    num_bytes: 2605620484
    num_examples: 1967
  download_size: 699214351
  dataset_size: 24696572552
- config_name: atari-surround
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22455378308
    num_examples: 17015
  - name: test
    num_bytes: 2959264212
    num_examples: 2243
  download_size: 904891011
  dataset_size: 25414642520
- config_name: atari-tennis
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22268080624
    num_examples: 16820
  - name: test
    num_bytes: 2454987940
    num_examples: 1855
  download_size: 924600351
  dataset_size: 24723068564
- config_name: atari-timepilot
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22188074736
    num_examples: 16764
  - name: test
    num_bytes: 2563118724
    num_examples: 1935
  download_size: 841644131
  dataset_size: 24751193460
- config_name: atari-tutankham
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 20778986696
    num_examples: 15686
  - name: test
    num_bytes: 3505680584
    num_examples: 2646
  download_size: 1008077517
  dataset_size: 24284667280
- config_name: atari-upndown
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21471544696
    num_examples: 16202
  - name: test
    num_bytes: 2655773200
    num_examples: 2004
  download_size: 4294716609
  dataset_size: 24127317896
- config_name: atari-venture
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 23569633620
    num_examples: 17883
  - name: test
    num_bytes: 2303843304
    num_examples: 1750
  download_size: 1281634956
  dataset_size: 25873476924
- config_name: atari-videopinball
  features:
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22728603252
    num_examples: 17155
  - name: test
    num_bytes: 856644644
    num_examples: 647
  download_size: 1494196602
  dataset_size: 23585247896
- config_name: atari-wizardofwor
  features:
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22489242716
    num_examples: 17025
  - name: test
    num_bytes: 2611112092
    num_examples: 1977
  download_size: 1717410677
  dataset_size: 25100354808
- config_name: atari-yarsrevenge
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22130113268
    num_examples: 16707
  - name: test
    num_bytes: 2590529456
    num_examples: 1956
  download_size: 3463955089
  dataset_size: 24720642724
- config_name: atari-zaxxon
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22175124172
    num_examples: 16757
  - name: test
    num_bytes: 2646613536
    num_examples: 2000
  download_size: 1227979546
  dataset_size: 24821737708
- config_name: babyai-action-obj-door
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 3397145228
    num_examples: 160519
  - name: test
    num_bytes: 179818964
    num_examples: 8497
  download_size: 37473537
  dataset_size: 3576964192
- config_name: babyai-blocked-unlock-pickup
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 14020875464
    num_examples: 666122
  - name: test
    num_bytes: 741637820
    num_examples: 35235
  download_size: 120761763
  dataset_size: 14762513284
- config_name: babyai-boss-level
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 35513875740
    num_examples: 1669851
  - name: test
    num_bytes: 1827605892
    num_examples: 85933
  download_size: 451555885
  dataset_size: 37341481632
- config_name: babyai-boss-level-no-unlock
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 34852472684
    num_examples: 1638751
  - name: test
    num_bytes: 1788394376
    num_examples: 84090
  download_size: 441749451
  dataset_size: 36640867060
- config_name: babyai-find-obj-s5
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 12575228852
    num_examples: 597321
  - name: test
    num_bytes: 655719752
    num_examples: 31146
  download_size: 128927098
  dataset_size: 13230948604
- config_name: babyai-go-to
  features:
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21423213864
    num_examples: 1007298
  - name: test
    num_bytes: 1110444816
    num_examples: 52212
  download_size: 253558881
  dataset_size: 22533658680
- config_name: babyai-go-to-door
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 3047640596
    num_examples: 143297
  - name: test
    num_bytes: 160467060
    num_examples: 7545
  download_size: 24299398
  dataset_size: 3208107656
- config_name: babyai-go-to-local
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 3050086416
    num_examples: 143412
  - name: test
    num_bytes: 161168904
    num_examples: 7578
  download_size: 32464811
  dataset_size: 3211255320
- config_name: babyai-go-to-obj
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2873736308
    num_examples: 135609
  - name: test
    num_bytes: 150255080
    num_examples: 7090
  download_size: 22694850
  dataset_size: 3023991388
- config_name: babyai-go-to-obj-door
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 3297170132
    num_examples: 155761
  - name: test
    num_bytes: 173221196
    num_examples: 8183
  download_size: 38908561
  dataset_size: 3470391328
- config_name: babyai-go-to-red-ball
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 3042231536
    num_examples: 143628
  - name: test
    num_bytes: 159437324
    num_examples: 7527
  download_size: 31219139
  dataset_size: 3201668860
- config_name: babyai-go-to-red-ball-grey
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 3260044112
    num_examples: 153284
  - name: test
    num_bytes: 173185324
    num_examples: 8143
  download_size: 28671074
  dataset_size: 3433229436
- config_name: babyai-go-to-red-ball-no-dists
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2879091696
    num_examples: 135372
  - name: test
    num_bytes: 151172944
    num_examples: 7108
  download_size: 18947867
  dataset_size: 3030264640
- config_name: babyai-go-to-red-blue-ball
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 3246068948
    num_examples: 153329
  - name: test
    num_bytes: 170342552
    num_examples: 8046
  download_size: 34540015
  dataset_size: 3416411500
- config_name: babyai-go-to-seq
  features:
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 30405239280
    num_examples: 1429644
  - name: test
    num_bytes: 1602763924
    num_examples: 75361
  download_size: 370458257
  dataset_size: 32008003204
- config_name: babyai-key-corridor
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 42831456224
    num_examples: 2037272
  download_size: 395532284
  dataset_size: 42831456224
- config_name: babyai-mini-boss-level
  features:
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 11850888756
    num_examples: 557225
  - name: test
    num_bytes: 618700476
    num_examples: 29091
  download_size: 139874065
  dataset_size: 12469589232
- config_name: babyai-move-two-across-s8n9
  features:
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 18403455616
    num_examples: 865312
  - name: test
    num_bytes: 967608928
    num_examples: 45496
  download_size: 271012706
  dataset_size: 19371064544
- config_name: babyai-one-room-s8
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 3231396116
    num_examples: 151937
  - name: test
    num_bytes: 169803712
    num_examples: 7984
  download_size: 21784870
  dataset_size: 3401199828
- config_name: babyai-open
  features:
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 12785992596
    num_examples: 601185
  - name: test
    num_bytes: 662051572
    num_examples: 31129
  download_size: 17470900
  dataset_size: 13448044168
- config_name: babyai-open-door
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 3836148388
    num_examples: 180373
  - name: test
    num_bytes: 199279112
    num_examples: 9370
  download_size: 33859673
  dataset_size: 4035427500
- config_name: babyai-open-doors-order-n4
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 5125013764
    num_examples: 240973
  - name: test
    num_bytes: 272251668
    num_examples: 12801
  download_size: 44783888
  dataset_size: 5397265432
- config_name: babyai-open-red-door
  features:
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2689734224
    num_examples: 126852
  - name: test
    num_bytes: 141009800
    num_examples: 6650
  download_size: 11432745
  dataset_size: 2830744024
- config_name: babyai-open-two-doors
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 6495523684
    num_examples: 305413
  - name: test
    num_bytes: 340692092
    num_examples: 16019
  download_size: 48013278
  dataset_size: 6836215776
- config_name: babyai-pickup
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 21481606304
    num_examples: 1021192
  - name: test
    num_bytes: 1108654424
    num_examples: 52702
  download_size: 257938606
  dataset_size: 22590260728
- config_name: babyai-pickup-above
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 12142108496
    num_examples: 576708
  - name: test
    num_bytes: 647924300
    num_examples: 30775
  download_size: 131363085
  dataset_size: 12790032796
- config_name: babyai-pickup-loc
  features:
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 3345296772
    num_examples: 157293
  - name: test
    num_bytes: 173418248
    num_examples: 8154
  download_size: 39790997
  dataset_size: 3518715020
- config_name: babyai-synth
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 20798549336
    num_examples: 977950
  - name: test
    num_bytes: 1075149172
    num_examples: 50553
  download_size: 254626142
  dataset_size: 21873698508
- config_name: babyai-synth-loc
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 19075051748
    num_examples: 906661
  - name: test
    num_bytes: 1011073416
    num_examples: 48058
  download_size: 244550361
  dataset_size: 20086125164
- config_name: babyai-synth-seq
  features:
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 32147938656
    num_examples: 1528824
  - name: test
    num_bytes: 1689908868
    num_examples: 80365
  download_size: 414173930
  dataset_size: 33837847524
- config_name: babyai-unblock-pickup
  features:
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 22543101672
    num_examples: 1059954
  - name: test
    num_bytes: 1208660440
    num_examples: 56830
  download_size: 272254758
  dataset_size: 23751762112
- config_name: babyai-unlock-local
  features:
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 6275032796
    num_examples: 297483
  - name: test
    num_bytes: 329361368
    num_examples: 15614
  download_size: 50666416
  dataset_size: 6604394164
- config_name: babyai-unlock-pickup
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 8595079376
    num_examples: 404132
  - name: test
    num_bytes: 453263616
    num_examples: 21312
  download_size: 82024594
  dataset_size: 9048342992
- config_name: metaworld-assembly
  features:
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2023926684
    num_examples: 95163
  - name: test
    num_bytes: 202067268
    num_examples: 9501
  download_size: 15003359
  dataset_size: 2225993952
- config_name: metaworld-basketball
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2027393368
    num_examples: 95326
  - name: test
    num_bytes: 200174416
    num_examples: 9412
  download_size: 73368828
  dataset_size: 2227567784
- config_name: metaworld-bin-picking
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2031200340
    num_examples: 95505
  - name: test
    num_bytes: 201620640
    num_examples: 9480
  download_size: 69672564
  dataset_size: 2232820980
- config_name: metaworld-box-close
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2028159016
    num_examples: 95362
  - name: test
    num_bytes: 201535568
    num_examples: 9476
  download_size: 72877607
  dataset_size: 2229694584
- config_name: metaworld-button-press
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2030222012
    num_examples: 95459
  - name: test
    num_bytes: 201301620
    num_examples: 9465
  download_size: 37777915
  dataset_size: 2231523632
- config_name: metaworld-button-press-topdown
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2031264144
    num_examples: 95508
  - name: test
    num_bytes: 201726980
    num_examples: 9485
  download_size: 40293880
  dataset_size: 2232991124
- config_name: metaworld-button-press-topdown-wall
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2033412212
    num_examples: 95609
  - name: test
    num_bytes: 203917584
    num_examples: 9588
  download_size: 42876573
  dataset_size: 2237329796
- config_name: metaworld-button-press-wall
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2030349620
    num_examples: 95465
  - name: test
    num_bytes: 202046000
    num_examples: 9500
  download_size: 39492416
  dataset_size: 2232395620
- config_name: metaworld-coffee-button
  features:
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2030009332
    num_examples: 95449
  - name: test
    num_bytes: 204810840
    num_examples: 9630
  download_size: 36937669
  dataset_size: 2234820172
- config_name: metaworld-coffee-pull
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2031136536
    num_examples: 95502
  - name: test
    num_bytes: 203109400
    num_examples: 9550
  download_size: 71581988
  dataset_size: 2234245936
- config_name: metaworld-coffee-push
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2029626508
    num_examples: 95431
  - name: test
    num_bytes: 202620236
    num_examples: 9527
  download_size: 76299775
  dataset_size: 2232246744
- config_name: metaworld-dial-turn
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2020247320
    num_examples: 94990
  - name: test
    num_bytes: 205427612
    num_examples: 9659
  download_size: 38454551
  dataset_size: 2225674932
- config_name: metaworld-disassemble
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2033880108
    num_examples: 95631
  - name: test
    num_bytes: 204279140
    num_examples: 9605
  download_size: 34969761
  dataset_size: 2238159248
- config_name: metaworld-door-close
  features:
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2034262932
    num_examples: 95649
  - name: test
    num_bytes: 203279544
    num_examples: 9558
  download_size: 55319560
  dataset_size: 2237542476
- config_name: metaworld-door-lock
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2029775384
    num_examples: 95438
  - name: test
    num_bytes: 204470552
    num_examples: 9614
  download_size: 42331037
  dataset_size: 2234245936
- config_name: metaworld-door-open
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2026776596
    num_examples: 95297
  - name: test
    num_bytes: 204959716
    num_examples: 9637
  download_size: 56814526
  dataset_size: 2231736312
- config_name: metaworld-door-unlock
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2030285816
    num_examples: 95462
  - name: test
    num_bytes: 203960120
    num_examples: 9590
  download_size: 40955468
  dataset_size: 2234245936
- config_name: metaworld-drawer-close
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2026159824
    num_examples: 95268
  - name: test
    num_bytes: 201429228
    num_examples: 9471
  download_size: 42045311
  dataset_size: 2227589052
- config_name: metaworld-drawer-open
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2034943508
    num_examples: 95681
  - name: test
    num_bytes: 202024732
    num_examples: 9499
  download_size: 34146052
  dataset_size: 2236968240
- config_name: metaworld-faucet-close
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2031668236
    num_examples: 95527
  - name: test
    num_bytes: 200195684
    num_examples: 9413
  download_size: 44921052
  dataset_size: 2231863920
- config_name: metaworld-faucet-open
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2020204784
    num_examples: 94988
  - name: test
    num_bytes: 204342944
    num_examples: 9608
  download_size: 43717974
  dataset_size: 2224547728
- config_name: metaworld-hammer
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2029031004
    num_examples: 95403
  - name: test
    num_bytes: 203343348
    num_examples: 9561
  download_size: 76772124
  dataset_size: 2232374352
- config_name: metaworld-hand-insert
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2027776192
    num_examples: 95344
  - name: test
    num_bytes: 203789976
    num_examples: 9582
  download_size: 58291877
  dataset_size: 2231566168
- config_name: metaworld-handle-press
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2026159824
    num_examples: 95268
  - name: test
    num_bytes: 204725768
    num_examples: 9626
  download_size: 47517332
  dataset_size: 2230885592
- config_name: metaworld-handle-press-side
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2024266972
    num_examples: 95179
  - name: test
    num_bytes: 202131072
    num_examples: 9504
  download_size: 43777749
  dataset_size: 2226398044
- config_name: metaworld-handle-pull
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2025628124
    num_examples: 95243
  - name: test
    num_bytes: 202407556
    num_examples: 9517
  download_size: 41337966
  dataset_size: 2228035680
- config_name: metaworld-handle-pull-side
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2026946740
    num_examples: 95305
  - name: test
    num_bytes: 201812052
    num_examples: 9489
  download_size: 51909803
  dataset_size: 2228758792
- config_name: metaworld-lever-pull
  features:
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2036942700
    num_examples: 95775
  - name: test
    num_bytes: 202322484
    num_examples: 9513
  download_size: 58372284
  dataset_size: 2239265184
- config_name: metaworld-peg-insert-side
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2025075156
    num_examples: 95217
  - name: test
    num_bytes: 202237412
    num_examples: 9509
  download_size: 59852799
  dataset_size: 2227312568
- config_name: metaworld-peg-unplug-side
  features:
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2024968816
    num_examples: 95212
  - name: test
    num_bytes: 203683636
    num_examples: 9577
  download_size: 78686871
  dataset_size: 2228652452
- config_name: metaworld-pick-out-of-hole
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2029307488
    num_examples: 95416
  - name: test
    num_bytes: 201663176
    num_examples: 9482
  download_size: 12886968
  dataset_size: 2230970664
- config_name: metaworld-pick-place
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2034199128
    num_examples: 95646
  - name: test
    num_bytes: 201812052
    num_examples: 9489
  download_size: 75007510
  dataset_size: 2236011180
- config_name: metaworld-pick-place-wall
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2024394580
    num_examples: 95185
  - name: test
    num_bytes: 203343348
    num_examples: 9561
  download_size: 75656602
  dataset_size: 2227737928
- config_name: metaworld-plate-slide
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2030328352
    num_examples: 95464
  - name: test
    num_bytes: 200387096
    num_examples: 9422
  download_size: 37916460
  dataset_size: 2230715448
- config_name: metaworld-plate-slide-back
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2025798268
    num_examples: 95251
  - name: test
    num_bytes: 202747844
    num_examples: 9533
  download_size: 14191177
  dataset_size: 2228546112
- config_name: metaworld-plate-slide-back-side
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2036985236
    num_examples: 95777
  - name: test
    num_bytes: 204108996
    num_examples: 9597
  download_size: 16325280
  dataset_size: 2241094232
- config_name: metaworld-plate-slide-side
  features:
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2023543860
    num_examples: 95145
  - name: test
    num_bytes: 201663176
    num_examples: 9482
  download_size: 27470169
  dataset_size: 2225207036
- config_name: metaworld-push
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2031455556
    num_examples: 95517
  - name: test
    num_bytes: 203173204
    num_examples: 9553
  download_size: 71989800
  dataset_size: 2234628760
- config_name: metaworld-push-back
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2034752096
    num_examples: 95672
  - name: test
    num_bytes: 203896316
    num_examples: 9587
  download_size: 53733439
  dataset_size: 2238648412
- config_name: metaworld-push-wall
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2038686676
    num_examples: 95857
  - name: test
    num_bytes: 204002656
    num_examples: 9592
  download_size: 71959178
  dataset_size: 2242689332
- config_name: metaworld-reach
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2029732848
    num_examples: 95436
  - name: test
    num_bytes: 200535972
    num_examples: 9429
  download_size: 53185239
  dataset_size: 2230268820
- config_name: metaworld-reach-wall
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2039239644
    num_examples: 95883
  - name: test
    num_bytes: 203556028
    num_examples: 9571
  download_size: 55080374
  dataset_size: 2242795672
- config_name: metaworld-shelf-place
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2023458788
    num_examples: 95141
  - name: test
    num_bytes: 202258680
    num_examples: 9510
  download_size: 58081868
  dataset_size: 2225717468
- config_name: metaworld-soccer
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2028265356
    num_examples: 95367
  - name: test
    num_bytes: 201599372
    num_examples: 9479
  download_size: 71952345
  dataset_size: 2229864728
- config_name: metaworld-stick-pull
  features:
  - name: loss_mask
    sequence: bool
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2030902588
    num_examples: 95491
  - name: test
    num_bytes: 204364212
    num_examples: 9609
  download_size: 68549865
  dataset_size: 2235266800
- config_name: metaworld-stick-push
  features:
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2023309912
    num_examples: 95134
  - name: test
    num_bytes: 203641100
    num_examples: 9575
  download_size: 70293105
  dataset_size: 2226951012
- config_name: metaworld-sweep
  features:
  - name: input_types
    sequence: int64
  - name: local_positions
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2031753308
    num_examples: 95531
  - name: test
    num_bytes: 203385884
    num_examples: 9563
  download_size: 73893717
  dataset_size: 2235139192
- config_name: metaworld-sweep-into
  features:
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2031136536
    num_examples: 95502
  - name: test
    num_bytes: 201918392
    num_examples: 9494
  download_size: 51708252
  dataset_size: 2233054928
- config_name: metaworld-window-close
  features:
  - name: loss_mask
    sequence: bool
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2031306680
    num_examples: 95510
  - name: test
    num_bytes: 204194068
    num_examples: 9601
  download_size: 38988729
  dataset_size: 2235500748
- config_name: metaworld-window-open
  features:
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: local_positions
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2025266568
    num_examples: 95226
  - name: test
    num_bytes: 202428824
    num_examples: 9518
  download_size: 38119386
  dataset_size: 2227695392
- config_name: mujoco-ant
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 7452041212
    num_examples: 354547
  - name: test
    num_bytes: 823589952
    num_examples: 39184
  download_size: 515277596
  dataset_size: 8275631164
- config_name: mujoco-doublependulum
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2780552408
    num_examples: 132222
  - name: test
    num_bytes: 308102556
    num_examples: 14651
  download_size: 170131756
  dataset_size: 3088654964
- config_name: mujoco-halfcheetah
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 5102353556
    num_examples: 242721
  - name: test
    num_bytes: 563902388
    num_examples: 26825
  download_size: 354562700
  dataset_size: 5666255944
- config_name: mujoco-hopper
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 1595264464
    num_examples: 75812
  - name: test
    num_bytes: 175096852
    num_examples: 8321
  download_size: 106662406
  dataset_size: 1770361316
- config_name: mujoco-humanoid
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 75816138508
    num_examples: 3564839
  - name: test
    num_bytes: 8423048000
    num_examples: 396048
  download_size: 4711062854
  dataset_size: 84239186508
- config_name: mujoco-pendulum
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 721246752
    num_examples: 34216
  - name: test
    num_bytes: 80752204
    num_examples: 3831
  download_size: 40182940
  dataset_size: 801998956
- config_name: mujoco-pusher
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 891111600
    num_examples: 42300
  - name: test
    num_bytes: 99411628
    num_examples: 4719
  download_size: 43780876
  dataset_size: 990523228
- config_name: mujoco-reacher
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 239130252
    num_examples: 11271
  - name: test
    num_bytes: 26689096
    num_examples: 1258
  download_size: 11606836
  dataset_size: 265819348
- config_name: mujoco-standup
  features:
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: input_types
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 83481551704
    num_examples: 3925278
  - name: test
    num_bytes: 9266536880
    num_examples: 435708
  download_size: 5334776801
  dataset_size: 92748088584
- config_name: mujoco-swimmer
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 2362203232
    num_examples: 112312
  - name: test
    num_bytes: 264523924
    num_examples: 12577
  download_size: 162879465
  dataset_size: 2626727156
- config_name: mujoco-walker
  features:
  - name: input_ids
    sequence: int32
  - name: input_types
    sequence: int64
  - name: loss_mask
    sequence: bool
  - name: local_positions
    sequence: int64
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 4710980972
    num_examples: 224095
  - name: test
    num_bytes: 527257204
    num_examples: 25081
  download_size: 327273365
  dataset_size: 5238238176
- config_name: ok-vqa
  features:
  - name: patch_positions
    sequence:
      sequence:
        sequence: float64
  - name: input_types
    sequence: int64
  - name: input_ids
    sequence: int32
  - name: loss_mask
    sequence: bool
  - name: patches
    sequence:
      sequence:
        sequence:
          sequence: uint8
  - name: attention_mask
    sequence: bool
  splits:
  - name: train
    num_bytes: 4509830776
    num_examples: 9009
  - name: test
    num_bytes: 2532677440
    num_examples: 5046
  download_size: 3512871870
  dataset_size: 7042508216
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: train
    path: atari-doubledunk/train-*
  - split: test
    path: atari-doubledunk/test-*
- 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: train
    path: atari-pong/train-*
  - split: test
    path: atari-pong/test-*
- config_name: atari-privateeye
  data_files:
  - split: train
    path: atari-privateeye/train-*
  - split: test
    path: atari-privateeye/test-*
- config_name: atari-qbert
  data_files:
  - split: train
    path: atari-qbert/train-*
  - split: test
    path: atari-qbert/test-*
- config_name: atari-riverraid
  data_files:
  - split: train
    path: atari-riverraid/train-*
  - split: test
    path: atari-riverraid/test-*
- config_name: atari-roadrunner
  data_files:
  - split: train
    path: atari-roadrunner/train-*
  - split: test
    path: atari-roadrunner/test-*
- config_name: atari-robotank
  data_files:
  - split: train
    path: atari-robotank/train-*
  - split: test
    path: atari-robotank/test-*
- config_name: atari-seaquest
  data_files:
  - split: train
    path: atari-seaquest/train-*
  - split: test
    path: atari-seaquest/test-*
- 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: train
    path: atari-venture/train-*
  - split: test
    path: atari-venture/test-*
- config_name: atari-videopinball
  data_files:
  - split: train
    path: atari-videopinball/train-*
  - split: test
    path: atari-videopinball/test-*
- config_name: atari-wizardofwor
  data_files:
  - split: train
    path: atari-wizardofwor/train-*
  - split: test
    path: atari-wizardofwor/test-*
- config_name: atari-yarsrevenge
  data_files:
  - split: train
    path: atari-yarsrevenge/train-*
  - split: test
    path: atari-yarsrevenge/test-*
- config_name: atari-zaxxon
  data_files:
  - split: train
    path: atari-zaxxon/train-*
  - split: test
    path: atari-zaxxon/test-*
- 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: train
    path: babyai-blocked-unlock-pickup/train-*
  - split: test
    path: babyai-blocked-unlock-pickup/test-*
- config_name: babyai-boss-level
  data_files:
  - split: train
    path: babyai-boss-level/train-*
  - split: test
    path: babyai-boss-level/test-*
- config_name: babyai-boss-level-no-unlock
  data_files:
  - split: train
    path: babyai-boss-level-no-unlock/train-*
  - split: test
    path: babyai-boss-level-no-unlock/test-*
- 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-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: train
    path: babyai-key-corridor/train-*
- config_name: babyai-mini-boss-level
  data_files:
  - split: train
    path: babyai-mini-boss-level/train-*
  - split: test
    path: babyai-mini-boss-level/test-*
- config_name: babyai-move-two-across-s8n9
  data_files:
  - split: train
    path: babyai-move-two-across-s8n9/train-*
  - split: test
    path: babyai-move-two-across-s8n9/test-*
- config_name: babyai-one-room-s8
  data_files:
  - split: train
    path: babyai-one-room-s8/train-*
  - split: test
    path: babyai-one-room-s8/test-*
- config_name: babyai-open
  data_files:
  - split: train
    path: babyai-open/train-*
  - split: test
    path: babyai-open/test-*
- config_name: babyai-open-door
  data_files:
  - split: train
    path: babyai-open-door/train-*
  - split: test
    path: babyai-open-door/test-*
- config_name: babyai-open-doors-order-n4
  data_files:
  - split: train
    path: babyai-open-doors-order-n4/train-*
  - split: test
    path: babyai-open-doors-order-n4/test-*
- config_name: babyai-open-red-door
  data_files:
  - split: train
    path: babyai-open-red-door/train-*
  - split: test
    path: babyai-open-red-door/test-*
- config_name: babyai-open-two-doors
  data_files:
  - split: train
    path: babyai-open-two-doors/train-*
  - split: test
    path: babyai-open-two-doors/test-*
- config_name: babyai-pickup
  data_files:
  - split: train
    path: babyai-pickup/train-*
  - split: test
    path: babyai-pickup/test-*
- config_name: babyai-pickup-above
  data_files:
  - split: train
    path: babyai-pickup-above/train-*
  - split: test
    path: babyai-pickup-above/test-*
- config_name: babyai-pickup-loc
  data_files:
  - split: train
    path: babyai-pickup-loc/train-*
  - split: test
    path: babyai-pickup-loc/test-*
- config_name: babyai-synth
  data_files:
  - split: train
    path: babyai-synth/train-*
  - split: test
    path: babyai-synth/test-*
- config_name: babyai-synth-loc
  data_files:
  - split: train
    path: babyai-synth-loc/train-*
  - split: test
    path: babyai-synth-loc/test-*
- config_name: babyai-synth-seq
  data_files:
  - split: train
    path: babyai-synth-seq/train-*
  - split: test
    path: babyai-synth-seq/test-*
- config_name: babyai-unblock-pickup
  data_files:
  - split: train
    path: babyai-unblock-pickup/train-*
  - split: test
    path: babyai-unblock-pickup/test-*
- config_name: babyai-unlock-local
  data_files:
  - split: train
    path: babyai-unlock-local/train-*
  - split: test
    path: babyai-unlock-local/test-*
- config_name: babyai-unlock-pickup
  data_files:
  - split: train
    path: babyai-unlock-pickup/train-*
  - split: test
    path: babyai-unlock-pickup/test-*
- 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-*
---
# Dataset Card for "gia-dataset-tokenized-1024"

[More Information needed](https://github.com/huggingface/datasets/blob/main/CONTRIBUTING.md#how-to-contribute-to-the-dataset-cards)