File size: 1,044 Bytes
de0f8c8 532d162 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
---
license: mit
---
This is the official model repository of the GenieRedux models, introduced by our work "[Learning Generative Interactive Environments By Trained Agent Exploration](https://openreview.net/forum?id=4AlHrKpRG7)" (NeurIPS'24 D3S3 Workshop). GenieRedux is an open implementation of the Genie world model, while GenieRedux-G is its variant, adapted for virtual environments.
To download and use the models, please follow the instructions in our [official code repository](https://github.com/insait-institute/GenieRedux).
Our models are trained on the CoinRun environment, following the case study of the original work. We provide:
- Tokenizer - a spatial-temporal tokenizer with a VQ codebook;
- GenieRedux - Dynamics+LAM, with action embedding from the LAM;
- GenieRedux-G - Dynamics only, with one-hot action input.
This model was created by INSAIT Institute, part of Sofia University, in Sofia, Bulgaria. |