Torch Profiler Memory. I fristly use the argument on_trace_ready to generate a tenso