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

Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch

MIT License

1.01k stars 37 forks source link

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

Open njkrichardson opened 9 months ago

njkrichardson commented 9 months ago

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?

Stonesjtu commented 9 months ago

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.