Hi, I’m working on extending the Graphs plugin for PyTorch models and want to study how the color gradient feature for Compute time and Memory works. Can someone please share with me a super simple example of this feature in action? It’d be really nice if it’s code I can run in Google Colab, and the model in the example can be a Tensorflow model. I Googled but couldn’t find a working example, unfortunately.
Simple example for the colored Compute time and Memory under Graphs tab?
Hi @Olivia_Liu, You can use the TensorFlow Profiler to profile the execution of your TensorFlow code. Please refer to this document for working code example. Thank You.