Giters

Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

KeyError: "['XXX'] not in index"

njkrichardson opened this issue · comments

commented

I added the @profile decorator to a custom module method (after from pytorch_memlab import LineProfiler, profile) and the output is the truncated display with the error KeyError: "['XXX'] not in index" for my method XXX. What's the meaning of this?

commented

Can you post the code snippet and the full error message? The error message indicates the function under profiling is missing. However I still need more information to find out the actual reason.