TF2.0中的Tensorflow分析

时间:2019-06-25 14:11:18

标签: tensorflow tensorboard tensorflow2.0

我正在尝试使用TF2.0(Beta)可视化tf.data.Datasets的性能。我找到了有关如何在旧版本的tensorflow中使用事件探查器的示例。 在TF2.0中如何进行性能分析?我可以使用tf.compat.v1,但是该过程似乎并不简单。

我想测量内存消耗(明智的设备放置)和时间轴。

以下示例说明了如何使用TF1.x进行配置 Can I measure the execution time of individual operations with TensorFlow?

Understanding tensorflow profiling results

1 个答案:

答案 0 :(得分:0)

与此同时,我找到了解决问题的方法: 如here所述,在我的训练步骤中使用trace_on和trace_export获取分析器输出 Example Profile