Hi All,
I am a beginner of pytorch, I tried to print memory cost (or variable shape/size) of each layer, not only the model._params and model._buffers, but also some mid-variable in model.backward(), how can I do it?
Thanks