本文整理匯總了Python中tensorflow.contrib.summary.always_record_summaries方法的典型用法代碼示例。如果您正苦於以下問題:Python summary.always_record_summaries方法的具體用法?Python summary.always_record_summaries怎麽用?Python summary.always_record_summaries使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類tensorflow.contrib.summary
的用法示例。
在下文中一共展示了summary.always_record_summaries方法的1個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: record_stats
# 需要導入模塊: from tensorflow.contrib import summary [as 別名]
# 或者: from tensorflow.contrib.summary import always_record_summaries [as 別名]
def record_stats(*, stats, summary_writer, step, log_interval, name=None, comm=MPI.COMM_WORLD):
def log_stats(step, *stat_values):
if comm.Get_rank() != 0 or step % log_interval != 0:
return
for k, v in safe_zip(stats.keys(), stat_values):
print('k = ', k, ', v = ', v)
summary_ops = [tf.py_func(log_stats, [step] + list(stats.values()), [])]
if summary_writer:
with summary_writer.as_default(), summary.always_record_summaries():
for key, value in stats.items():
summary_ops.append(summary.scalar(key, value, step=step))
return tf.group(*summary_ops, name=name)