def summary(value_dict, global_step, writer):
"""Make tf.Summary for tensorboard"""
summary = tf.Summary(value=[tf.Summary.Value(tag=k, simple_value=v) for k, v in value_dict.items()])
writer.add_summary(summary, global_step)
return None
评论列表
文章目录