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

Update world_model.py shape #23

Open suryadheeshjith opened 9 months ago

suryadheeshjith commented 9 months ago

Updating comment here due to shape error