Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to get memory diff between first and second batch?

khursani8 opened this issue · comments

commented

I'm using MemReporter
After run reporter.report()
I don't know which part different since there are lot of layer and tensor being output
Wondering if there is a way to find diff easily

Good idea, I'm trying to add a special symbol (like *) along the tensor name for newly added tensors.

Stay tuned.