相关疑难解决方法(0)

我正在尝试将一些TensorBoard日志记录添加到使用新的tf.estimator API的模型中.

我有一个钩子设置如下:

summary_hook = tf.train.SummarySaverHook(
    save_secs=2,
    output_dir=MODEL_DIR,
    summary_op=tf.summary.merge_all())

# ...

classifier.train(
    input_fn,
    steps=1000,
    hooks=[summary_hook])

在我model_fn,我也创造了summary-

def model_fn(features, labels, mode):
    # ... model stuff, calculate the value of loss
    tf.summary.scalar("loss", loss)
    # ...

但是,当我运行此代码时,我从以下错误中得到以下错误summary_hook: Exactly one of scaffold or summary_op must be provided.这可能是因为tf.summary.merge_all()找不到任何摘要并且正在返回None,尽管tf.summary.scalar我已在声明中声明了model_fn.

任何想法为什么这不起作用？