episode_hooks.py 文件源码

def after_run(self, run_context, run_values):
        global_episode = run_values.results['global_episode']
        if can_run_hook(run_context):
            if self._timer.should_trigger_for_episode(global_episode):
                original = np.get_printoptions()
                np.set_printoptions(suppress=True)
                elapsed_secs, _ = self._timer.update_last_triggered_episode(global_episode)
                if self._formatter:
                    logging.info(self._formatter(run_values.results))
                else:
                    stats = []
                    for tag in self._tag_order:
                        stats.append("%s = %s" % (tag, run_values.results[tag]))
                    if elapsed_secs is not None:
                        logging.info("%s (%.3f sec)", ", ".join(stats), elapsed_secs)
                    else:
                        logging.info("%s", ", ".join(stats))
                np.set_printoptions(**original)