Giters

Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Memory leak when calling mem reporter

Stonesjtu opened this issue · comments

commented

The tensor_name and tensor_device mapping dict maintains a reference to the key / value tensor, which prevents python collecting unused tensors.

commented

Fixed in #18