如何报告算法运行时间？

Question

我正在TensorFlow中运行可变自动编码器，这可能需要很长时间。 因此，我想报告算法在TensorBoard上作为标量运行的时间。

一种肮脏的方法是将编译的开始时间硬编码为全局变量，或者将其作为参数传递给模型函数，然后计算与当前时间的差。

Tensorflow是否有本地方法？

Answer 1

有。带有1.14版。

用法示例：

estimator = tf.estimator.LinearClassifier(...)
hooks = [tf.train.ProfilerHook(output_dir=model_dir, save_secs=600, show_memory=False)]
estimator.train(input_fn=train_input_fn, hooks=hooks)

执行挂钩会在timeline-xx.json中生成文件output_dir。

然后在Chrome浏览器中打开chrome://tracing/并加载文件。您将获得如下的时间使用时间表。 1