I am trying to profile some standard networks. However when I try to use the pytorch profiling scripts, the report only lists the computational time of some pytorch built-in functions. They are not differentiated by where they are called. Is there a way to append a name to these functions or is there a way to track them by where they are called?
I also tried a third party ‘line profiler’ but that tool is giving weird/incorrect profiling results.