torchprof

PyTorch layer-by-layer model profiler

Archived

GitHub

608 stars
11 watching
47 forks
Language: Python
last commit: over 3 years ago

Screenshot of awwong1/torchprof website

pypi.org/project/torchprof/