使用带有tf.Estimator的Tensorflow分析器

Question

我需要使用Tensorflow分析器来分析一些由于某种原因而运行缓慢的代码.不幸的是,有问题的代码使用tf.Estimator,所以我无法弄清楚如何注入运行元数据对象到会话中的run()调用,以获取信息的探查需求.

我该怎么办？

Answer 1

使用 ProfileContext，如下所述： https: //github.com/tensorflow/tensorflow/tree/master/tensorflow/core/profiler。这使您无需访问会话即可进行分析。