使用TensorBoard进行性能分析时，只会生成"Profile“选项卡中提供的5个工具中的一个(trace_viewer

Question

我想分析一下CNN，看看哪些层占用了更多的执行时间。

首先，我导入在Keras中创建的模型，并加载我已经训练好的参数

model = my_model.model_arch(parameters)    

model.load_weights(weights_path)

然后我启用跟踪

logdir = 'logs' 

writer = tf.summary.create_file_writer(logdir)

tf.summary.trace_on(graph=True, profiler=True)

然后我计算预测：

prediction_1=model.predict(x=input,steps=steps_1)

最后，我写下了轨迹：

with writer.as_default():
    tf.summary.trace_export(name="model_trace", step=0, profiler_outdir='logs/trace')

但是，在"profile“选项卡中，除了"trace_viewer”之外，我看不到任何其他内容。特别是，我对"TensorFlow统计“工具(如https://www.tensorflow.org/guide/profiler中所示)感兴趣。我还尝试了一种不同的回调方法：

logdir = 'logs' 

model = my_model.model_arch(parameters)    

model.load_weights(weights_path)

tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=logdir, histogram_freq=1, profile_batch = '1', write_graph=True, write_images=True)

prediction_1=model.predict(x=input,steps=steps_1,callbacks=[tensorboard_callback])

但它甚至没有生成配置文件选项卡，只生成了一个"graphs“选项卡。

有什么建议吗？

Answer 1

这很可能是由于使用较旧的TensorFlow版本(< 2.2.0)造成的。以前的版本没有所有这些功能。