Imagination-augmented agent plays Sokoban

An agent plays Sokoban from a pixel representation, not knowing the rules of the game. At specific points in time, we visualise the agent’s imagina...