python - 如何从 tf.estimator 获取默认 session ？

Question

我正在尝试高级 API tf.estimator，但我发现很难让 session 调试某些内部结果，例如全局步骤。

cls = tf.estimator.Estimator(
    model_fn=my_model,
    params={
        'feature_columns': fcs,
        'hidden_units': [10, 10],
        'n_classes': 3,
    })

例子来自 https://www.tensorflow.org/versions/master/get_started/custom_estimators

我尝试了 sess = tf.get_default_session 和 with tf.Session() as sess，但无法获得 defut session 。

Answer 1

最简单的方法是像这样使用 tf.Print:

...
global_step = tf.Print(global_step, [global_step], message='Value of global step")
...

您可以将 global_step 替换为您想要打印的任何张量。然后，当您运行训练时，它会在每次评估张量时打印值。

另一种更复杂的方法是导出模型，然后使用您自己的 session (而不是估算器 API)将其重新加载。执行此操作后，您可以为定义的任何操作调用 session.run。您可以使用 tf.get_operation_by_name 或 tf.get_tensor_by_name 获取操作。您还可以提供任何您想要的值作为输入。