Tensorflow Estimator API中的摘要 - 需要更清晰

Question

有人可以帮助我使用Estimator API创建有关丢失和准确性的张量板摘要。我在model_func中创建了tf.summary.scalar('loss',loss)和tf.summary.scalar('accuracy',accuracy)。但是，我只看到张量板上的global_step摘要。有些机构可以指出我使用estimator API

来解释摘要编写器和日志记录钩子的相关示例

Answer 1

1）在model_fn中创建摘要 2）将它们与tf.summary.merge_all合并 3）将合并的摘要传递给tf.train.SummarySaverHook
4）将此挂钩传递给您使用training_hooks List的{​​{1}}参数（作为EstimatorSpec） >

注意：model_fn中定义的摘要不能传递到函数的scpoe之外。因此，您无法使用model_fn方法的hooks参数传递您的摘要。

Answer 2

您需要将这些摘要与估算工具挂钩。像这样：

  # tensors_to_log can be either a list or a dictionary.
  tensors_to_log = ['loss', 'accuracy']
  logging_hook = tf.train.LoggingTensorHook(tensors=tensors_to_log, every_n_iter=100)
  xx_estimator.train(input_fn,steps, hooks=[logging_hook])

可以在此tensorflow mnist example中找到更完整的示例。