rlu_atari

  • Description:

RL Unplugged is suite of benchmarks for offline reinforcement learning. The RL Unplugged is designed around the following considerations: to facilitate ease of use, we provide the datasets with a unified API which makes it easy for the practitioner to work with all data in the suite once a general pipeline has been established.

The datasets follow the RLDS format to represent steps and episodes.

We are releasing a large and diverse dataset of gameplay following the protocol described by Agarwal et al., 2020, which can be used to evaluate several discrete offline RL algorithms. The dataset is generated by running an online DQN agent and recording transitions from its replay during training with sticky actions Machado et al., 2018. As stated in Agarwal et al., 2020, for each game we use data from five runs with 50 million transitions each. We release datasets for 46 Atari games. For details on how the dataset was generated, please refer to the paper.

Atari is a standard RL benchmark. We recommend you to try offline RL methods on Atari if you are interested in comparing your approach to other state of the art offline RL methods with discrete actions.

The reward of each step is clipped (obtained with [-1, 1] clipping) and the episode includes the sum of the clipped reward per episode.

FeaturesDict({
    'checkpoint_id': tf.int64,
    'episode_id': tf.int64,
    'episode_return': Scalar(shape=(), dtype=tf.float32),
    'steps': Dataset({
        'action': tf.int64,
        'discount': tf.float32,
        'is_first': tf.bool,
        'is_last': tf.bool,
        'is_terminal': tf.bool,
        'observation': Image(shape=(84, 84, 1), dtype=tf.uint8),
        'reward': Scalar(shape=(), dtype=tf.float32),
    }),
})
  • Feature documentation:
Feature Class Shape Dtype Description Value range
FeaturesDict
checkpoint_id Tensor tf.int64
episode_id Tensor tf.int64
episode_return Scalar tf.float32 Sum of the clipped rewards.
steps Dataset
steps/action Tensor tf.int64
steps/discount Tensor tf.float32
steps/is_first Tensor tf.bool
steps/is_last Tensor tf.bool
steps/is_terminal Tensor tf.bool
steps/observation Image (84, 84, 1) tf.uint8
steps/reward Scalar tf.float32 Clipped reward. [-1, 1]
@misc{gulcehre2020rl,
    title={RL Unplugged: Benchmarks for Offline Reinforcement Learning},
    author={Caglar Gulcehre and Ziyu Wang and Alexander Novikov and Tom Le Paine
        and  Sergio Gómez Colmenarejo and Konrad Zolna and Rishabh Agarwal and
        Josh Merel and Daniel Mankowitz and Cosmin Paduraru and Gabriel
        Dulac-Arnold and Jerry Li and Mohammad Norouzi and Matt Hoffman and
        Ofir Nachum and George Tucker and Nicolas Heess and Nando deFreitas},
    year={2020},
    eprint={2006.13888},
    archivePrefix={arXiv},
    primaryClass={cs.LG}
}

rlu_atari/Alien_run_1 (default config)

  • Dataset size: 73.90 GiB

  • Splits:

Split Examples
'train' 55,945

rlu_atari/Alien_run_2

  • Dataset size: 74.13 GiB

  • Splits:

Split Examples
'train' 50,353

rlu_atari/Alien_run_3

  • Dataset size: 74.02 GiB

  • Splits:

Split Examples
'train' 53,194

rlu_atari/Alien_run_4

  • Dataset size: 74.43 GiB

  • Splits:

Split Examples
'train' 55,405

rlu_atari/Alien_run_5

  • Dataset size: 74.09 GiB

  • Splits:

Split Examples
'train' 54,836

rlu_atari/Amidar_run_1

  • Dataset size: 76.91 GiB

  • Splits:

Split Examples
'train' 23,153

rlu_atari/Amidar_run_2

  • Dataset size: 76.70 GiB

  • Splits:

Split Examples
'train' 22,713

rlu_atari/Amidar_run_3

  • Dataset size: 77.03 GiB

  • Splits:

Split Examples
'train' 28,050

rlu_atari/Amidar_run_4

  • Dataset size: 77.16 GiB

  • Splits:

Split Examples
'train' 26,716

rlu_atari/Amidar_run_5

  • Dataset size: 77.42 GiB

  • Splits:

Split Examples
'train' 26,027

rlu_atari/Assault_run_1

  • Dataset size: 28.70 GiB

  • Splits:

Split Examples
'train' 44,237

rlu_atari/Assault_run_2

  • Dataset size: 28.62 GiB

  • Splits:

Split Examples
'train' 46,114

rlu_atari/Assault_run_3

  • Dataset size: 28.66 GiB

  • Splits:

Split Examples
'train' 46,146

rlu_atari/Assault_run_4

  • Dataset size: 28.64 GiB

  • Splits:

Split Examples
'train' 45,517

rlu_atari/Assault_run_5

  • Dataset size: 28.63 GiB

  • Splits:

Split Examples
'train' 45,283

rlu_atari/Asterix_run_1

  • Dataset size: 46.55 GiB

  • Splits:

Split Examples
'train' 57,870

rlu_atari/Asterix_run_2

  • Dataset size: 46.64 GiB

  • Splits:

Split Examples
'train' 56,851

rlu_atari/Asterix_run_3

  • Dataset size: 46.45 GiB

  • Splits:

Split Examples
'train' 67,577

rlu_atari/Asterix_run_4

  • Dataset size: 46.57 GiB

  • Splits:

Split Examples
'train' 58,131

rlu_atari/Asterix_run_5

  • Dataset size: 46.60 GiB

  • Splits:

Split Examples
'train' 58,081

rlu_atari/Atlantis_run_1

  • Dataset size: 41.46 GiB

  • Splits:

Split Examples
'train' 9,937

rlu_atari/Atlantis_run_2

  • Dataset size: 41.25 GiB

  • Splits:

Split Examples
'train' 10,736

rlu_atari/Atlantis_run_3

  • Dataset size: 41.48 GiB

  • Splits:

Split Examples
'train' 9,056

rlu_atari/Atlantis_run_4

  • Dataset size: 41.54 GiB

  • Splits:

Split Examples
'train' 10,255

rlu_atari/Atlantis_run_5

  • Dataset size: 41.48 GiB

  • Splits:

Split Examples
'train' 9,689

rlu_atari/BankHeist_run_1

  • Dataset size: 71.93 GiB

  • Splits:

Split Examples
'train' 41,824

rlu_atari/BankHeist_run_2

  • Dataset size: 71.93 GiB

  • Splits:

Split Examples
'train' 42,097

rlu_atari/BankHeist_run_3

  • Dataset size: 71.82 GiB

  • Splits:

Split Examples
'train' 43,758

rlu_atari/BankHeist_run_4

  • Dataset size: 71.93 GiB

  • Splits:

Split Examples
'train' 42,234

rlu_atari/BankHeist_run_5

  • Dataset size: 71.96 GiB

  • Splits:

Split Examples
'train' 41,935

rlu_atari/BattleZone_run_1

  • Dataset size: 57.82 GiB

  • Splits:

Split Examples
'train' 23,864

rlu_atari/BattleZone_run_2

  • Dataset size: 57.79 GiB

  • Splits:

Split Examples
'train' 26,224

rlu_atari/BattleZone_run_3

  • Dataset size: 57.80 GiB

  • Splits:

Split Examples
'train' 24,706

rlu_atari/BattleZone_run_4

  • Dataset size: 57.73 GiB

  • Splits:

Split Examples
'train' 22,650

rlu_atari/BattleZone_run_5

  • Dataset size: 57.83 GiB

  • Splits:

Split Examples
'train' 25,447

rlu_atari/BeamRider_run_1

  • Dataset size: 43.72 GiB

  • Splits:

Split Examples
'train' 16,403

rlu_atari/BeamRider_run_2

  • Dataset size: 43.74 GiB

  • Splits:

Split Examples
'train' 16,138

rlu_atari/BeamRider_run_3

  • Dataset size: 43.72 GiB

  • Splits:

Split Examples
'train' 16,606

rlu_atari/BeamRider_run_4

  • Dataset size: 43.72 GiB

  • Splits:

Split Examples
'train' 16,316

rlu_atari/BeamRider_run_5

  • Dataset size: 43.74 GiB

  • Splits:

Split Examples
'train' 16,076

rlu_atari/Boxing_run_1

  • Dataset size: 45.81 GiB

  • Splits:

Split Examples
'train' 49,485

rlu_atari/Boxing_run_2

  • Dataset size: 45.59 GiB

  • Splits:

Split Examples
'train' 53,651

rlu_atari/Boxing_run_3

  • Dataset size: 45.79 GiB

  • Splits:

Split Examples
'train' 53,000

rlu_atari/Boxing_run_4

  • Dataset size: 45.68 GiB

  • Splits:

Split Examples
'train' 53,756

rlu_atari/Boxing_run_5

  • Dataset size: 45.79 GiB

  • Splits:

Split Examples
'train' 51,467

rlu_atari/Breakout_run_1

  • Dataset size: 30.66 GiB

  • Splits:

Split Examples
'train' 41,839

rlu_atari/Breakout_run_2

  • Dataset size: 30.62 GiB

  • Splits:

Split Examples
'train' 42,291

rlu_atari/Breakout_run_3

  • Dataset size: 30.65 GiB

  • Splits:

Split Examples
'train' 41,774

rlu_atari/Breakout_run_4

  • Dataset size: 30.66 GiB

  • Splits:

Split Examples
'train' 41,691

rlu_atari/Breakout_run_5

  • Dataset size: 30.71 GiB

  • Splits:

Split Examples
'train' 41,703

rlu_atari/Carnival_run_1

  • Dataset size: 33.58 GiB

  • Splits:

Split Examples
'train' 218,526

rlu_atari/Carnival_run_2

  • Dataset size: 33.69 GiB

  • Splits:

Split Examples
'train' 222,221

rlu_atari/Carnival_run_3

  • Dataset size: 33.59 GiB

  • Splits:

Split Examples
'train' 224,169

rlu_atari/Carnival_run_4

  • Dataset size: 33.60 GiB

  • Splits:

Split Examples
'train' 218,607

rlu_atari/Carnival_run_5

  • Dataset size: 34.51 GiB

  • Splits:

Split Examples
'train' 218,428

rlu_atari/Centipede_run_1

  • Dataset size: 35.78 GiB

  • Splits:

Split Examples
'train' 94,369

rlu_atari/Centipede_run_2

  • Dataset size: 35.67 GiB

  • Splits:

Split Examples
'train' 105,430

rlu_atari/Centipede_run_3

  • Dataset size: 35.68 GiB

  • Splits:

Split Examples
'train' 105,965

rlu_atari/Centipede_run_4

  • Dataset size: 35.94 GiB

  • Splits:

Split Examples
'train' 99,675

rlu_atari/Centipede_run_5

  • Dataset size: 35.95 GiB

  • Splits:

Split Examples
'train' 86,569

rlu_atari/ChopperCommand_run_1

  • Dataset size: 40.01 GiB

  • Splits:

Split Examples
'train' 69,258

rlu_atari/ChopperCommand_run_2

  • Dataset size: 40.16 GiB

  • Splits:

Split Examples
'train' 66,438

rlu_atari/ChopperCommand_run_3

  • Dataset size: 40.16 GiB

  • Splits:

Split Examples
'train' 68,073

rlu_atari/ChopperCommand_run_4

  • Dataset size: 40.24 GiB

  • Splits:

Split Examples
'train' 67,685

rlu_atari/ChopperCommand_run_5

  • Dataset size: 40.33 GiB

  • Splits:

Split Examples
'train' 67,438

rlu_atari/CrazyClimber_run_1

  • Dataset size: 51.22 GiB

  • Splits:

Split Examples
'train' 12,547

rlu_atari/CrazyClimber_run_2

  • Dataset size: 51.68 GiB

  • Splits:

Split Examples
'train' 12,833

rlu_atari/CrazyClimber_run_3

  • Dataset size: 51.19 GiB

  • Splits:

Split Examples
'train' 12,587

rlu_atari/CrazyClimber_run_4

  • Dataset size: 51.11 GiB

  • Splits:

Split Examples
'train' 12,414

rlu_atari/CrazyClimber_run_5

  • Dataset size: 51.30 GiB

  • Splits:

Split Examples
'train' 12,479

rlu_atari/DemonAttack_run_1

  • Dataset size: 24.09 GiB

  • Splits:

Split Examples
'train' 22,438

rlu_atari/DemonAttack_run_2

  • Dataset size: 24.23 GiB

  • Splits:

Split Examples
'train' 19,770

rlu_atari/DemonAttack_run_3

  • Dataset size: 24.22 GiB

  • Splits:

Split Examples
'train' 19,817

rlu_atari/DemonAttack_run_4

  • Dataset size: 24.02 GiB

  • Splits:

Split Examples
'train' 23,391

rlu_atari/DemonAttack_run_5

  • Dataset size: 23.86 GiB

  • Splits:

Split Examples
'train' 25,903

rlu_atari/DoubleDunk_run_1

  • Dataset size: 49.40 GiB

  • Splits:

Split Examples
'train' 14,393

rlu_atari/DoubleDunk_run_2

  • Dataset size: 49.37 GiB

  • Splits:

Split Examples
'train' 15,077

rlu_atari/DoubleDunk_run_3

  • Dataset size: 49.33 GiB

  • Splits:

Split Examples
'train' 14,357

rlu_atari/DoubleDunk_run_4

  • Dataset size: 49.36 GiB

  • Splits:

Split Examples
'train' 14,700

rlu_atari/DoubleDunk_run_5

  • Dataset size: 49.38 GiB

  • Splits:

Split Examples
'train' 14,718

rlu_atari/Enduro_run_1

  • Dataset size: 56.76 GiB

  • Splits:

Split Examples
'train' 5,634

rlu_atari/Enduro_run_2

  • Dataset size: 56.75 GiB

  • Splits:

Split Examples
'train' 5,642

rlu_atari/Enduro_run_3

  • Dataset size: 56.76 GiB

  • Splits:

Split Examples
'train' 5,769

rlu_atari/Enduro_run_4

  • Dataset size: 56.76 GiB

  • Splits:

Split Examples
'train' 5,601

rlu_atari/Enduro_run_5

  • Dataset size: 56.74 GiB

  • Splits:

Split Examples
'train' 5,775

rlu_atari/FishingDerby_run_1

  • Dataset size: 63.57 GiB

  • Splits:

Split Examples
'train' 28,353

rlu_atari/FishingDerby_run_2

  • Dataset size: 63.57 GiB

  • Splits:

Split Examples
'train' 27,962

rlu_atari/FishingDerby_run_3

  • Dataset size: 63.56 GiB

  • Splits:

Split Examples
'train' 28,533

rlu_atari/FishingDerby_run_4

  • Dataset size: 63.59 GiB

  • Splits:

Split Examples
'train' 27,786

rlu_atari/FishingDerby_run_5

  • Dataset size: 63.58 GiB

  • Splits:

Split Examples
'train' 27,811

rlu_atari/Freeway_run_1

  • Dataset size: 50.94 GiB

  • Splits:

Split Examples
'train' 24,428

rlu_atari/Freeway_run_2

  • Dataset size: 50.95 GiB

  • Splits:

Split Examples
'train' 24,428

rlu_atari/Freeway_run_3

  • Dataset size: 50.99 GiB

  • Splits:

Split Examples
'train' 24,428

rlu_atari/Freeway_run_4

  • Dataset size: 51.00 GiB

  • Splits:

Split Examples
'train' 24,428

rlu_atari/Freeway_run_5

  • Dataset size: 50.54 GiB

  • Splits:

Split Examples
'train' 24,428

rlu_atari/Frostbite_run_1

  • Dataset size: 57.03 GiB

  • Splits:

Split Examples
'train' 67,091

rlu_atari/Frostbite_run_2

  • Dataset size: 56.26 GiB

  • Splits:

Split Examples
'train' 69,165

rlu_atari/Frostbite_run_3

  • Dataset size: 56.57 GiB

  • Splits:

Split Examples
'train' 65,693

rlu_atari/Frostbite_run_4

  • Dataset size: 56.79 GiB

  • Splits:

Split Examples
'train' 66,150

rlu_atari/Frostbite_run_5

  • Dataset size: 56.58 GiB

  • Splits:

Split Examples
'train' 65,228

rlu_atari/Gopher_run_1

  • Dataset size: 36.31 GiB

  • Splits:

Split Examples
'train' 24,272

rlu_atari/Gopher_run_2

  • Dataset size: 36.35 GiB

  • Splits:

Split Examples
'train' 23,591

rlu_atari/Gopher_run_3

  • Dataset size: 36.33 GiB

  • Splits:

Split Examples
'train' 24,238

rlu_atari/Gopher_run_4

  • Dataset size: 36.33 GiB

  • Splits:

Split Examples
'train' 23,415

rlu_atari/Gopher_run_5

  • Dataset size: 36.32 GiB

  • Splits:

Split Examples
'train' 24,136

rlu_atari/Gravitar_run_1

  • Dataset size: 33.91 GiB

  • Splits:

Split Examples
'train' 75,393

rlu_atari/Gravitar_run_2

  • Dataset size: 33.67 GiB

  • Splits:

Split Examples
'train' 74,719

rlu_atari/Gravitar_run_3

  • Dataset size: 33.11 GiB

  • Splits:

Split Examples
'train' 74,262

rlu_atari/Gravitar_run_4

  • Dataset size: 33.16 GiB

  • Splits:

Split Examples
'train' 74,766

rlu_atari/Gravitar_run_5

  • Dataset size: 33.48 GiB

  • Splits:

Split Examples
'train' 74,043

rlu_atari/Hero_run_1

  • Dataset size: 46.65 GiB

  • Splits:

Split Examples
'train' 41,461

rlu_atari/Hero_run_2

  • Dataset size: 46.45 GiB

  • Splits:

Split Examples
'train' 36,680

rlu_atari/Hero_run_3

  • Dataset size: 46.66 GiB

  • Splits:

Split Examples
'train' 40,925

rlu_atari/Hero_run_4

  • Dataset size: 46.67 GiB

  • Splits:

Split Examples
'train' 40,757

rlu_atari/Hero_run_5

  • Dataset size: 46.74 GiB

  • Splits:

Split Examples
'train' 43,907

rlu_atari/IceHockey_run_1

  • Dataset size: 37.09 GiB

  • Splits:

Split Examples
'train' 15,623

rlu_atari/IceHockey_run_2

  • Dataset size: 37.09 GiB

  • Splits:

Split Examples
'train' 15,610

rlu_atari/IceHockey_run_3

  • Dataset size: 37.18 GiB

  • Splits:

Split Examples
'train' 15,132

rlu_atari/IceHockey_run_4

  • Dataset size: 37.16 GiB

  • Splits:

Split Examples
'train' 15,342

rlu_atari/IceHockey_run_5

  • Dataset size: 37.18 GiB

  • Splits:

Split Examples
'train' 15,287

rlu_atari/Jamesbond_run_1

  • Dataset size: 35.02 GiB

  • Splits:

Split Examples
'train' 34,791

rlu_atari/Jamesbond_run_2

  • Dataset size: 35.04 GiB

  • Splits:

Split Examples
'train' 37,727

rlu_atari/Jamesbond_run_3

  • Dataset size: 35.19 GiB

  • Splits:

Split Examples
'train' 33,891

rlu_atari/Jamesbond_run_4

  • Dataset size: 34.99 GiB

  • Splits:

Split Examples
'train' 35,252

rlu_atari/Jamesbond_run_5

  • Dataset size: 35.11 GiB

  • Splits:

Split Examples
'train' 33,579

rlu_atari/Kangaroo_run_1

  • Dataset size: 55.07 GiB

  • Splits:

Split Examples
'train' 25,748

rlu_atari/Kangaroo_run_2

  • Dataset size: 54.60 GiB

  • Splits:

Split Examples
'train' 26,428

rlu_atari/Kangaroo_run_3

  • Dataset size: 54.54 GiB

  • Splits:

Split Examples
'train' 28,684

rlu_atari/Kangaroo_run_4

  • Dataset size: 55.21 GiB

  • Splits:

Split Examples
'train' 24,495

rlu_atari/Kangaroo_run_5

  • Dataset size: 54.37 GiB

  • Splits:

Split Examples
'train' 25,568

rlu_atari/Krull_run_1

  • Dataset size: 90.36 GiB

  • Splits:

Split Examples
'train' 37,989

rlu_atari/Krull_run_2

  • Dataset size: 90.41 GiB

  • Splits:

Split Examples
'train' 38,011

rlu_atari/Krull_run_3

  • Dataset size: 90.27 GiB

  • Splits:

Split Examples
'train' 38,204

rlu_atari/Krull_run_4

  • Dataset size: 90.38 GiB

  • Splits:

Split Examples
'train' 37,960

rlu_atari/Krull_run_5

  • Dataset size: 90.39 GiB

  • Splits:

Split Examples
'train' 38,109

rlu_atari/KungFuMaster_run_1

  • Dataset size: 61.89 GiB

  • Splits:

Split Examples
'train' 19,460

rlu_atari/KungFuMaster_run_2

  • Dataset size: 61.74 GiB

  • Splits:

Split Examples
'train' 21,367

rlu_atari/KungFuMaster_run_3

  • Dataset size: 61.77 GiB

  • Splits:

Split Examples
'train' 20,591

rlu_atari/KungFuMaster_run_4

  • Dataset size: 62.00 GiB

  • Splits:

Split Examples
'train' 18,700

rlu_atari/KungFuMaster_run_5

  • Dataset size: 61.99 GiB

  • Splits:

Split Examples
'train' 18,580

rlu_atari/MsPacman_run_1

  • Dataset size: 75.55 GiB

  • Splits:

Split Examples
'train' 41,975

rlu_atari/MsPacman_run_2

  • Dataset size: 75.64 GiB

  • Splits:

Split Examples
'train' 42,282

rlu_atari/MsPacman_run_3

  • Dataset size: 75.75 GiB

  • Splits:

Split Examples
'train' 41,289

rlu_atari/MsPacman_run_4

  • Dataset size: 75.39 GiB

  • Splits:

Split Examples
'train' 41,983

rlu_atari/MsPacman_run_5

  • Dataset size: 75.43 GiB

  • Splits:

Split Examples
'train' 42,831

rlu_atari/NameThisGame_run_1

  • Dataset size: 60.81 GiB

  • Splits:

Split Examples
'train' 11,303

rlu_atari/NameThisGame_run_2

  • Dataset size: 60.87 GiB

  • Splits:

Split Examples
'train' 11,506

rlu_atari/NameThisGame_run_3

  • Dataset size: 60.68 GiB

  • Splits:

Split Examples
'train' 11,260

rlu_atari/NameThisGame_run_4

  • Dataset size: 60.97 GiB

  • Splits:

Split Examples
'train' 12,259

rlu_atari/NameThisGame_run_5

  • Dataset size: 61.11 GiB

  • Splits:

Split Examples
'train' 12,483

rlu_atari/Phoenix_run_1

  • Dataset size: 27.14 GiB

  • Splits:

Split Examples
'train' 30,241

rlu_atari/Phoenix_run_2

  • Dataset size: 27.27 GiB

  • Splits:

Split Examples
'train' 27,471

rlu_atari/Phoenix_run_3

  • Dataset size: 27.63 GiB

  • Splits:

Split Examples
'train' 27,417

rlu_atari/Phoenix_run_4

  • Dataset size: 27.22 GiB

  • Splits:

Split Examples
'train' 30,163

rlu_atari/Phoenix_run_5

  • Dataset size: 27.16 GiB

  • Splits:

Split Examples
'train' 31,181

rlu_atari/Pong_run_1

  • Dataset size: 20.86 GiB

  • Splits:

Split Examples
'train' 23,239

rlu_atari/Pong_run_2

  • Dataset size: 21.13 GiB

  • Splits:

Split Examples
'train' 16,862

rlu_atari/Pong_run_3

  • Dataset size: 21.09 GiB

  • Splits:

Split Examples
'train' 18,593

rlu_atari/Pong_run_4

  • Dataset size: 21.11 GiB

  • Splits:

Split Examples
'train' 17,112

rlu_atari/Pong_run_5

  • Dataset size: 20.90 GiB

  • Splits:

Split Examples
'train' 23,268

rlu_atari/Pooyan_run_1

  • Dataset size: 55.36 GiB

  • Splits:

Split Examples
'train' 29,471

rlu_atari/Pooyan_run_2

  • Dataset size: 55.28 GiB

  • Splits:

Split Examples
'train' 29,940

rlu_atari/Pooyan_run_3

  • Dataset size: 55.26 GiB

  • Splits:

Split Examples
'train' 30,167

rlu_atari/Pooyan_run_4

  • Dataset size: 55.38 GiB

  • Splits:

Split Examples
'train' 29,314

rlu_atari/Pooyan_run_5

  • Dataset size: 55.33 GiB

  • Splits:

Split Examples
'train' 29,811

rlu_atari/Qbert_run_1

  • Dataset size: 96.76 GiB

  • Splits:

Split Examples
'train' 47,635

rlu_atari/Qbert_run_2

  • Dataset size: 97.08 GiB

  • Splits:

Split Examples
'train' 46,382

rlu_atari/Qbert_run_3

  • Dataset size: 97.26 GiB

  • Splits:

Split Examples
'train' 46,775

rlu_atari/Qbert_run_4

  • Dataset size: 97.13 GiB

  • Splits:

Split Examples
'train' 47,412

rlu_atari/Qbert_run_5

  • Dataset size: 96.93 GiB

  • Splits:

Split Examples
'train' 46,099

rlu_atari/Riverraid_run_1

  • Dataset size: 48.46 GiB

  • Splits:

Split Examples
'train' 37,536

rlu_atari/Riverraid_run_2

  • Dataset size: 48.62 GiB

  • Splits:

Split Examples
'train' 37,998

rlu_atari/Riverraid_run_3

  • Dataset size: 48.60 GiB

  • Splits:

Split Examples
'train' 38,086

rlu_atari/Riverraid_run_4

  • Dataset size: 47.73 GiB

  • Splits:

Split Examples
'train' 64,952

rlu_atari/Riverraid_run_5

  • Dataset size: 48.49 GiB

  • Splits:

Split Examples
'train' 37,823

rlu_atari/RoadRunner_run_1

  • Dataset size: 41.72 GiB

  • Splits:

Split Examples
'train' 45,469

rlu_atari/RoadRunner_run_2

  • Dataset size: 41.66 GiB

  • Splits:

Split Examples
'train' 47,447

rlu_atari/RoadRunner_run_3

  • Dataset size: 41.70 GiB

  • Splits:

Split Examples
'train' 45,812

rlu_atari/RoadRunner_run_4

  • Dataset size: 41.70 GiB

  • Splits:

Split Examples
'train' 46,139

rlu_atari/RoadRunner_run_5

  • Dataset size: 41.73 GiB

  • Splits:

Split Examples
'train' 45,202

rlu_atari/Robotank_run_1

  • Dataset size: 47.20 GiB

  • Splits:

Split Examples
'train' 9,144

rlu_atari/Robotank_run_2

  • Dataset size: 47.28 GiB

  • Splits:

Split Examples
'train' 9,000

rlu_atari/Robotank_run_3

  • Dataset size: 47.31 GiB

  • Splits:

Split Examples
'train' 9,235

rlu_atari/Robotank_run_4

  • Dataset size: 47.35 GiB

  • Splits:

Split Examples
'train' 9,089

rlu_atari/Robotank_run_5

  • Dataset size: 47.13 GiB

  • Splits:

Split Examples
'train' 9,339

rlu_atari/Seaquest_run_1

  • Dataset size: 40.70 GiB

  • Splits:

Split Examples
'train' 33,872

rlu_atari/Seaquest_run_2

  • Dataset size: 41.63 GiB

  • Splits:

Split Examples
'train' 32,606

rlu_atari/Seaquest_run_3

  • Dataset size: 41.72 GiB

  • Splits:

Split Examples
'train' 35,662

rlu_atari/Seaquest_run_4

  • Dataset size: 41.59 GiB

  • Splits:

Split Examples
'train' 34,697

rlu_atari/Seaquest_run_5

  • Dataset size: 41.70 GiB

  • Splits:

Split Examples
'train' 32,783

rlu_atari/SpaceInvaders_run_1

  • Dataset size: 49.70 GiB

  • Splits:

Split Examples
'train' 30,333

rlu_atari/SpaceInvaders_run_2

  • Dataset size: 50.03 GiB

  • Splits:

Split Examples
'train' 32,534

rlu_atari/SpaceInvaders_run_3

  • Dataset size: 49.92 GiB

  • Splits:

Split Examples
'train' 30,449

rlu_atari/SpaceInvaders_run_4

  • Dataset size: 49.90 GiB

  • Splits:

Split Examples
'train' 32,587

rlu_atari/SpaceInvaders_run_5

  • Dataset size: 49.92 GiB

  • Splits:

Split Examples
'train' 32,337

rlu_atari/StarGunner_run_1

  • Dataset size: 22.65 GiB

  • Splits:

Split Examples
'train' 50,939

rlu_atari/StarGunner_run_2

  • Dataset size: 22.94 GiB

  • Splits:

Split Examples
'train' 41,624

rlu_atari/StarGunner_run_3

  • Dataset size: 22.85 GiB

  • Splits:

Split Examples
'train' 43,315

rlu_atari/StarGunner_run_4

  • Dataset size: 22.62 GiB

  • Splits:

Split Examples
'train' 50,397

rlu_atari/StarGunner_run_5

  • Dataset size: 22.75 GiB

  • Splits:

Split Examples
'train' 48,224

rlu_atari/TimePilot_run_1

  • Dataset size: 41.48 GiB

  • Splits:

Split Examples
'train' 47,855

rlu_atari/TimePilot_run_2

  • Dataset size: 41.38 GiB

  • Splits:

Split Examples
'train' 48,216

rlu_atari/TimePilot_run_3

  • Dataset size: 41.65 GiB

  • Splits:

Split Examples
'train' 46,181

rlu_atari/TimePilot_run_4

  • Dataset size: 41.68 GiB

  • Splits:

Split Examples
'train' 46,061

rlu_atari/TimePilot_run_5

  • Dataset size: 41.52 GiB

  • Splits:

Split Examples
'train' 47,624

rlu_atari/UpNDown_run_1

  • Dataset size: 100.18 GiB

  • Splits:

Split Examples
'train' 33,847

rlu_atari/UpNDown_run_2

  • Dataset size: 99.74 GiB

  • Splits:

Split Examples
'train' 33,467

rlu_atari/UpNDown_run_3

  • Dataset size: 99.85 GiB

  • Splits:

Split Examples
'train' 33,425

rlu_atari/UpNDown_run_4

  • Dataset size: 99.97 GiB

  • Splits:

Split Examples
'train' 33,018

rlu_atari/UpNDown_run_5

  • Dataset size: 100.06 GiB

  • Splits:

Split Examples
'train' 33,591

rlu_atari/VideoPinball_run_1

  • Dataset size: 71.74 GiB

  • Splits:

Split Examples
'train' 12,037

rlu_atari/VideoPinball_run_2

  • Dataset size: 71.43 GiB

  • Splits:

Split Examples
'train' 23,868

rlu_atari/VideoPinball_run_3

  • Dataset size: 71.04 GiB

  • Splits:

Split Examples
'train' 25,253

rlu_atari/VideoPinball_run_4

  • Dataset size: 71.72 GiB

  • Splits:

Split Examples
'train' 13,809

rlu_atari/VideoPinball_run_5

  • Dataset size: 71.74 GiB

  • Splits:

Split Examples
'train' 13,239

rlu_atari/WizardOfWor_run_1

  • Dataset size: 60.51 GiB

  • Splits:

Split Examples
'train' 37,887

rlu_atari/WizardOfWor_run_2

  • Dataset size: 60.47 GiB

  • Splits:

Split Examples
'train' 39,005

rlu_atari/WizardOfWor_run_3

  • Dataset size: 60.47 GiB

  • Splits:

Split Examples
'train' 47,518

rlu_atari/WizardOfWor_run_4

  • Dataset size: 60.10 GiB

  • Splits:

Split Examples
'train' 46,291

rlu_atari/WizardOfWor_run_5

  • Dataset size: 60.80 GiB

  • Splits:

Split Examples
'train' 44,904

rlu_atari/YarsRevenge_run_1

  • Dataset size: 69.21 GiB

  • Splits:

Split Examples
'train' 55,982

rlu_atari/YarsRevenge_run_2

  • Dataset size: 70.15 GiB

  • Splits:

Split Examples
'train' 56,225

rlu_atari/YarsRevenge_run_3

  • Dataset size: 70.41 GiB

  • Splits:

Split Examples
'train' 58,010

rlu_atari/YarsRevenge_run_4

  • Dataset size: 69.32 GiB

  • Splits:

Split Examples
'train' 56,292

rlu_atari/YarsRevenge_run_5

  • Dataset size: 71.13 GiB

  • Splits:

Split Examples
'train' 57,275

rlu_atari/Zaxxon_run_1

  • Dataset size: 38.87 GiB

  • Splits:

Split Examples
'train' 31,146

rlu_atari/Zaxxon_run_2

  • Dataset size: 39.03 GiB

  • Splits:

Split Examples
'train' 33,318

rlu_atari/Zaxxon_run_3

  • Dataset size: 38.83 GiB

  • Splits:

Split Examples
'train' 31,266

rlu_atari/Zaxxon_run_4

  • Dataset size: 39.13 GiB

  • Splits:

Split Examples
'train' 32,862

rlu_atari/Zaxxon_run_5

  • Dataset size: 37.57 GiB

  • Splits:

Split Examples
'train' 44,701