Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"
Primary LanguagePythonGNU General Public License v3.0GPL-3.0