Profiling and inspecting memory in pytorch - View it on GitHub