How to analyze model performance in TF?

PyTorch provides an API torch.profiler to analyze the model performance (e.g., execution time and memory allocation).

Is there a similar API in TesnforFlow?

Hi @shaoyu_young, TesorFlow also has a Profiler, which helps to understand the hardware resource consumption (time and memory) of the various TensorFlow operations (ops) in your model and resolve performance bottlenecks and, ultimately, make the model execute faster. Please refer to this document to know more. Thank You.

1 Like

thank u for your reply. :smiley: