Layer-by-layer profiler in `torch.profiler`

How to set up parameters to show layer-by-layer profile by torch.profiler?

The layer-by-layer profile is the same functionality in the package of torchprof, which shows the running time of each layers in NN. However, the current torch.profiler only displays low-level operations.

See Readme of torchprof for the difference of layer profiles and low-level operations, where the first table is layer-by-layer profile and the second table is low-level operations.


Same question. OP please share if you found a solution for this. Thank you!