ml-jku/mazerunner

--- license: mit task_categories: - reinforcement-learning --- # MazeRunner This repository contains the MazeRunner datasets as used in ["Retrieval-augmented Decision Transformer: External Memory for In-context RL"](https://huggingface.co/papers/2410.07071): - Datasets for grid-size 15x15. - The `15x15` folder contains 300 `.npz` files. Not all of them were used for our experiments. Download the dataset using: ``` huggingface-cli download ml-jku/mazerunner --local-dir=./mazerunner --repo-type dataset ``` For dataloading we refer to our Github repository: https://github.com/ml-jku/RA-DT Citation: ``` @article{schmied2024retrieval, title={Retrieval-Augmented Decision Transformer: External Memory for In-context RL}, author={Schmied, Thomas and Paischer, Fabian and Patil, Vihang and Hofmarcher, Markus and Pascanu, Razvan and Hochreiter, Sepp}, journal={arXiv preprint arXiv:2410.07071}, year={2024} } ```