在TensorFlow中记录通信

时间:2018-06-25 19:06:38

标签: python tensorflow

我有一个在TensorFlow中运行的NMT模型。它在2个GPU上运行。

我想记录每个计算操作所花费的时间,如果一个计算操作需要来自其他GPU或CPU的输入数据,我还想记录将所需的输入传送到指定的计算设备(例如GPU或GPU)所需的时间。其他计算设备的CPU。

我看着timeline。但是很难理解那个json。

我觉得可以从该json中提取所需的信息,但是,我没有找到任何文档来说明如何在时间轴中记录事件。

如何使用现有的日志记录API来做到这一点?

0 个答案:

没有答案