watch_atari

Utility to watch a trained agent play an Atari game.

watch_atari(saved_dir, env=None, num_episodes=10)

Watch a trained agent play an atari game.

Parameters

  • saved_dir (str) – Directory containing the pickle file.

  • env (str) – Environment to run episodes on. If None, the pickled environment is used.

  • num_episodes (int) – Number of episodes to play. Note that when using the EpisodicLife wrapper, an episode is considered done when a life is lost. Defaults to 10.