保存tf.summary.image Estimator API

此刻我想对TF Estimator API有点熟悉。我正在学习这个blog entry的例子。保存tf.summary.image Estimator API

现在我有问题，我无法保存简单的图像摘要（tf.summary.image）。

我在experiment_fn增加了SummarySaverHook和钩添加到eval_hooks参数实验构造：

def experiment_fn(run_config, params): 
    ... 
    summary_hook = tf.train.SummarySaverHook(
     100, 
     output_dir='/summeries', 
     summary_op=tf.summary.merge_all() 
    ) 
    ... 
    experiment = tf.contrib.learn.Experiment(
     estimator=estimator, # Estimator 
     train_input_fn=train_input_fn, # First-class function 
     eval_input_fn=eval_input_fn, # First-class function 
     train_steps=params.train_steps, # Minibatch steps 
     min_eval_frequency=params.min_eval_frequency, # Eval frequency 
     train_monitors=[train_input_hook], # Hooks for training 
     eval_hooks=[eval_input_hook, summary_hook], # Hooks for evaluation 
     eval_steps=None # Use evaluation feeder until its empty 
    )

的不是很清楚，我是在那里我有添加tf.summary.image（）函数使图像将显示在张量板的图像选项卡中。

在此先感谢。

来源

2017-10-17 Sebastian Metzler

你不应该添加一个钩子。只需在您的model_fn/input_fn中的任意位置添加tf.summary.image电话，估算器应自动为所有创建的摘要添加摘要挂钩。

来源

2017-10-18 00:02:23 DomJack

非常感谢。当我直接在我的model_fn中添加'tf.summary.image'时，所有内容都适用于训练模式。但是，当我试图保存仅用于评估模式的图像摘要时，摘要不会保存/显示在Tensorboard'f模式== ModeKeys.EVAL'中 –

保存tf.summary.image Estimator API

回答

相关问题