Swall0w / torchstat

Model analyzer in PyTorch
MIT License
1.47k stars 142 forks source link

[feature]: compute theoretical amount of memory read and write #3

Closed Swall0w closed 6 years ago

Swall0w commented 6 years ago

Like chainer-computational-cost, it is important to estimate memory read and write for each layer.