TF2.0 中的 Tensorflow 分析

yuv*_*blr 6 tensorflow tensorboard tensorflow2.0

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

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

下面的示例解释了使用 TF1.x 的分析 我可以使用 TensorFlow 测量单个操作的执行时间吗?

了解 tensorflow 分析结果

yuv*_*blr 3

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