轮廓张量流估计量

Question

我正忙着对tensorflow BERT code进行评测，看看它是否足够快以满足特定的用例。在

我需要确定预测的速度。现在，据我所知，可以在session.Run选项中添加profiler或run_metadata对象。但是，BERT正在使用新的EstimatorAPI。在

如何描述估计器？有可能吗？在

更新：

Tensorflow团队向我展示了here可以将ProfilerHook附加到一个估计器上：

hooks = [tf.train.ProfilerHook(save_steps=1, output_dir=model_dir)]
estimator.predict(..., hooks=hooks)

这将生成一个跟踪文件，但跟踪不起作用。在

^{pr2}$

Answer 1

如果你用过tf.估计器将创建timeline-.js类型的文件。将这些时间线-.js加载到chrome://追踪/查看跟踪文件 您将在model目录中找到timeline-.js文件。在