def build_summary_ops():
"""Tensorflow magic episode summary operations.
I have no idea what this does or how this works."""
episode_reward = tf.Variable(0.)
tf.scalar_summary("Episode Reward", episode_reward)
r_summary_placeholder = tf.placeholder("float")
update_ep_reward = episode_reward.assign(r_summary_placeholder)
ep_avg_v = tf.Variable(0.)
tf.scalar_summary("Episode Value", ep_avg_v)
val_summary_placeholder = tf.placeholder("float")
update_ep_val = ep_avg_v.assign(val_summary_placeholder)
summary_op = tf.merge_all_summaries()
return (r_summary_placeholder, update_ep_reward, val_summary_placeholder,
update_ep_val, summary_op)
评论列表
文章目录