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.