Is the pytorch profiler what you're thinking of? https://pytorch.org/tutorials/intermediate/tensorboard_profiler_tutorial.html https://pytorch.org/tutorials/in

Is the pytorch profiler what you're thinking of? https://pytorch.org/tutorials/intermediate/tensorboard_profiler_tutorial.html https://pytorch.org/tutorials/intermediate/tensorboard_profi...