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.
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.