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

804 stars 80 forks source link

Closed qiwang067 closed 9 months ago

qiwang067 commented 9 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 9 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!