This repository contains the Dark-Room datasets as used in "Retrieval-augmented Decision Transformer: External Memory for In-context RL":
Datasets for 3 grid-sizes: 10x10, 20x20, 40x20.
Every split contains 100 .npz files, corresponding to 100 start-goal position combinations.
Each file contains 100K transitions.
Filenames have the format START_GOAL.npz. E.g., for [0,0]_[0,5].npz, [0.0] is the start position and the goal is located at [0,5].
Download the dataset using:… See the full description on the dataset page:
https://huggingface.co/datasets/ml-jku/dark_room.
