Hello!
I’m working on making a inspector which examines each tensor, or nn.Module’s gpu/cpu memory resource consumption. (since nvidia-smi only shows total consumption)
Is there any built-in pytorch method to achieve this goal?
Thanks!