EpisodeHooks

[source]

EpisodeLoggingTensorHook

polyaxon.estimators.hooks.episode_hooks.EpisodeLoggingTensorHook(tensors, every_n_episodes, formatter=None)

[source]

StopAtEpisodeHook

polyaxon.estimators.hooks.episode_hooks.StopAtEpisodeHook(num_episodes=None, last_episode=None)

Monitor to request stop at a specified episode.


[source]

EpisodeSummarySaverHook

polyaxon.estimators.hooks.episode_hooks.EpisodeSummarySaverHook(save_episodes=None, output_dir=None, summary_writer=None, scaffold=None, summary_op=None)

Saves summaries every N episode.

  • Args:

    • save_episodes: int, save summaries every N episodes. Exactly one of save_secs and save_episodes should be set.
    • output_dir: string, the directory to save the summaries to. Only used if no summary_writer is supplied.
    • summary_writer: SummaryWriter. If None and an output_dir was passed, one will be created accordingly.
    • scaffold: Scaffold to get summary_op if it's not provided.
    • summary_op: Tensor of type string containing the serialized Summary protocol buffer or a list of Tensor. They are most likely an output by TF summary methods like tf.summary.scalar or tf.summary.merge_all. It can be passed in as one tensor; if more than one, they must be passed in as a list.
  • Raises:

    • ValueError: Exactly one of scaffold or summary_op should be set.

[source]

EpisodeCounterHook

polyaxon.estimators.hooks.episode_hooks.EpisodeCounterHook(output_dir=None, summary_writer=None)

TimeSteps and Seconds per episode.