Profiler UI
01 Feb 2019使用Profiler UI可视化分析Tensorflow耗时
https://github.com/tensorflow/profiler-ui
使用方法:
- 在model中加入with context,生成profile文件:
with tf.contrib.tfprof.ProfileContext(model_dir) as pctx:
tf.estimator.train_and_evaluate(model, train_spec, eval_spec)
- 使用profiler_ui在chrome中可视化profile文件
python profiler-ui/ui.py --profile_context_path=checkpoints/xxx/profile_100