如何从tf.estimator.Estimator中获取张量流会话?

时间:2019-04-02 10:23:04

标签: tensorflow

此文件记录在哪里?是否仍推荐估算器,或者只是将技术债务添加到您的学习曲线中?

想要深入了解传递给估算器的函数。 model_fn,input_fn等。

1 个答案:

答案 0 :(得分:1)

TensorFlow估算器已在TF 2.0中逐步淘汰。 Keras模型现在是推荐的高级API (youtube source)

我不确定您要如何处理会话,但是使用SessionRunHook

可以方便地访问它

您可以定义一个新的钩子,并将其传递给估算器,例如。

estimator.train(input_fn=train_input_fn, hooks=[my_custom_hook])