torchstat icon indicating copy to clipboard operation
torchstat copied to clipboard

Published 20 hours ago • verified publisher icon Swall0w

something about memory

Open NirHeaven opened this issue 3 years ago • 0 comments

Is the calculation of memory occupation the sum of the sizes of all feature maps? Or is it the calculation of peak memory ?

NirHeaven avatar May 27 '21 09:05 NirHeaven