How to reproduce the results of IRIS?

eloialonso / iris

Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.

https://openreview.net/forum?id=vhFu1Acb0xb

GNU General Public License v3.0

791 stars 77 forks source link

Closed qiwang067 closed 7 months ago

qiwang067 commented 7 months ago

Hi, Alonso, thank you for providing the detailed code. I want to reproduce the results of IRIS (https://github.com/eloialonso/iris/blob/main/results/data/IRIS.json), but didn't find the corresponding code.

eloialonso commented 7 months ago

Hi,

As explained in the README, you have to clone the repo, set up the dependencies, and then launch a training run to reproduce the results for a given game.

Hope that helps!