Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Tensorflow profiling in TF2.0

I am trying to visualize the performance of tf.data.Datasets using TF2.0 (Beta). I found examples on how to use profiler in older versions of tensorflow. How is profiling done in TF2.0? I could use tf.compat.v1, but the procedure does not seem to be straight forward.

I want to measure memory consumption (device placement wise) and timeline.

Below examples explain profiling with TF1.x Can I measure the execution time of individual operations with TensorFlow?

Understanding tensorflow profiling results

like image 915
yuva-rajulu Avatar asked Jun 25 '19 14:06

yuva-rajulu


1 Answers

Meanwhile, I found solution to my question: Using the trace_on and trace_export around my training step to get the profiler output, as described here Example Profile

like image 158
yuva-rajulu Avatar answered Oct 16 '22 15:10

yuva-rajulu