Smaller World Models for Reinforcement Learning

Benchmark Model Rank Results
atari-games-on-atari-2600-bank-heistDiscrete Latent Space World Model (VQ-VAE)–Score: 121.6
atari-games-on-atari-2600-breakoutDiscrete Latent Space World Model (VQ-VAE)–Score: 11.6
atari-games-on-atari-2600-crazy-climberDiscrete Latent Space World Model (VQ-VAE)–Score: 59609.4
atari-games-on-atari-2600-freewayDiscrete Latent Space World Model (VQ-VAE)–Score: 29
atari-games-on-atari-2600-pongDiscrete Latent Space World Model (VQ-VAE)–Score: 20.2
atari-games-on-atari-2600-seaquestDiscrete Latent Space World Model (VQ-VAE)–Score: 635