How to set up parameters to show layer-by-layer profile by
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.