torchprof

PyTorch layer-by-layer model profiler

Archived

GitHub

608 stars
11 watching
46 forks
Language: Python
last commit: almost 4 years ago

Screenshot of awwong1/torchprof website

pypi.org/project/torchprof/