search

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

IRIS agent in Hugging face transformers #26

Open RUFFY-369 opened 2 months ago

RUFFY-369 commented 2 months ago

Hi folks @eloialonso @vmicheli ,

First of all, a brilliant paper and I had great fun going through and forth the text and especially the code, every nitty-gritty. Inspired by the results, I decided to perform a little contribution to port the IRIS agent into Hugging face transformers library.

Here is the tutorial for training ported IRIS agent with Hugging face trainer: https://github.com/RUFFY-369/Transformers-tutorials/blob/main/IRIS/train_Iris_agent_on_atari.ipynb

Cheers!

vmicheli commented 2 months ago

Hey,

Thanks for the kind words and the work you have put in this port! We will make sure to add a pointer to your contribution and the notebook once the PR is merged.

Best, Vincent and Eloi

RUFFY-369 commented 2 months ago

Hi @vmicheli, Thank you for replying. It was my pleasure all along. The pointer will be a great appreciation token. :+1: I hope the PR gets merged soon. :)