See how to use RLDS in this tutorial.
You can find more examples in the following colabs:
- Performance best practices
- RL examples
- Colab for loading uniformly subsampled RLU Atari datasets such as 1%, 5% and 10% Atari datasets, as introduced by Agarwal et al. (2020) and commonly used in offline RL research (e.g., CQL, Decision Transformer, MuZero ReAnalyze).
- Generate and load a dataset with TFDS