tf.summary.imageをEstimator APIで保存します

Question

現在、私はTF Estimator APIをもう少し詳しく知ろうとしています。私はこのblog entryの例を使って作業/学習しています。

今、私は単純な画像の要約（tf.summary.image）を保存できないという問題があります。私は追加する必要がされてどのような私には明確ではありません

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 
    ) 

：

私はexperiment_fnにSummarySaverHookを追加し、実験コンストラクタのeval_hooksパラメータにフックを追加tf.summary.image（）テンソルボードの画像タブに画像が表示されるように機能します。

ありがとうございます。

Answer 1

フックを追加する必要はありません。 model_fn/input_fnのどこにでもtf.summary.imageコールを追加するだけで、エスティメータは自動的に作成されたすべてのサマリーのサマリーフックを追加する必要があります。