tf.train.SummarySaverHook用法

原创于 2019-01-31 10:19:36 发布 · 3.5k 阅读

0 ·

CC 4.0 BY-SA版权

tensorflow 专栏收录该内容

8 篇文章

订阅专栏

本文详细解析了TensorFlow中SummarySaverHook的用法，包括初始化参数的含义及如何配置，如save_steps和save_secs的使用场景，output_dir与summary_writer的关系，以及scaffold和summary_op的选择。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tf.train.SummarySaverHook用法

初始化函数如下

__init__(
    save_steps=None,
    save_secs=None,
    output_dir=None,
    summary_writer=None,
    scaffold=None,
    summary_op=None
)

用法说明：
1， save_steps和save_secs只需要传入一个。即每n个steps或每n秒输出一次summary。
2， output_dir和summary_writer只需要传入一个。如果传入了output_dir，会自动建立writer。
scaffold和summary_op只需要传入一个。当找不到summary_op时，会去使用scaffold.summary_op。