Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Variable 'tensor_names' referenced before assignment

sidmani opened this issue · comments

for param, name in tensor_names.items():

The variable tensor_names is referenced before assignment if no model is passed into the MemReporter. Need to move it into the above if statement.

Thanks for reporting this issue, will do this week.