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?
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.
thank u for your reply.