TensorFlow：RunMetadata中的成本图表为空

Question

我按照this answer获取了TensorFlow的成本模型估算值。但是，当我打印生成的d = {} l = [1,2,3,4,5] for i in range(len(l)): d[i] = l[i] 时，输出为空。还有什么我需要做的吗？

我希望看到TensorFlow成本模型的成本估算以及它如何映射到设备。还有其他办法可以实现这个目标吗？

这是完整的程序：

metadata.cost_graph

Answer 1

tf.GraphOptions.build_cost_model选项为defined，因为它是在生成成本模型之前运行的步骤数。从定义：

// The number of steps to run before returning a cost model detailing
// the memory usage and performance of each node of the graph. 0 means
// no cost model.
int64 build_cost_model = 4;

因此，要查看费用模型，您必须使用build_cost_model=50运行图表50次。为了获得更好的性能，我建议仅在第50次调用时传递run_options（否则TensorFlow将生成并丢弃大量元数据）。