Tensorflow：如何从时间线检查中获取内存消耗信息

Question

我试图从tensorflow run_meta选项的时间轴检查模型内存消耗，代码如下：

tl = timeline.Timeline(run_metadata.step_stats)
trace = tl.generate_chrome_trace_format()
trace_filename = os.path.join(train_step_kwargs['logdir'],
                              'tf_trace-%d.json' % np_global_step)
logging.info('Writing trace to %s', trace_filename)
file_io.write_string_to_file(trace_filename, trace)

对于输出的时间轴跟踪文件，我在chrome上得到了它：

但如何查看我的模型的内存消耗信息，我的意思是分层，我还没有找到任何有用的教程。