CODEWITHSUNDEEP
tensorflow
Konstantin Solomatov
Konstantin Solomatov

Reputation: 10332

How to profile TensorFlow model

How can I profile TensorFlow models? I want at least to find ways to make the model faster, see how much GPU load I have, and how much GPU memory is consumed.

Upvotes: 4

Views: 2923

Answers (1)

Salvador Dali
Salvador Dali

Reputation: 222511

There are two ways to profile models. One way is a tensorboard. Here is a comprehensive tutorial about it and here is a good video.

Additionally, clicking on a node will display the exact total memory, compute time, and tensor output sizes.

enter image description here

Another way is tensorflow debugger, which also has tutorials.

Upvotes: 6

Related Questions