This is an enhanced version of the TensorFlow Profiler (TFProf). With the enhanced TFProf, users are able to view multiple profile contexts on the same timeline page.
- Install Python dependencies.
pip install --user -r requirements.txt
- Install pprof.
- Create a profile context file using the tf.contrib.tfprof.ProfileContext class.
- Start the UI.
python ui.py --profile_context_path=/path/to/your/profile/dir --successive_profile_context_count=3
You can learn more about the TensorFlow Profiler's Python API and CLI here.
Currently only Chrome is supported.