TF2.0 中的 Tensorflow 分析

Tensorflow profiling in TF2.0

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

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

下面的示例解释了使用 TF1.x 进行分析

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