How to get memory diff between first and second batch?

Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch

MIT License

1.01k stars 37 forks source link

How to get memory diff between first and second batch? #48

Open khursani8 opened 1 year ago

khursani8 commented 1 year ago

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

Stonesjtu commented 1 year ago

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

Stay tuned.