albanD:

I’m afraid it’s not possible as some buffers are only saved on the cpp side and not accessible from python.


Is that still the case?
If yes, would there be a way to persist the computational graph to disk? In that case, we could measure the size from there.