保存tf.summary.image Estimator API

Question

此刻我想對TF Estimator API有點熟悉。我正在學習這個blog entry的例子。

現在我有問題，我無法保存簡單的圖像摘要（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（）函數使圖像將顯示在張量板的圖像選項卡中。

在此先感謝。

Answer 1

你不應該添加一個鉤子。只需在您的model_fn/input_fn中的任意位置添加tf.summary.image電話，估算器應自動爲所有創建的摘要添加摘要掛鉤。