Simple PyTorch profiler that combines DeepSpeed Flops Profiler and TorchInfo
Stars
11
Forks
0
Watchers
Open Issues
Overall repository health assessment
No package.json found
This might not be a Node.js project
User
5
commits
consider other module types that don't need input/output shape to be profiled
4af3171
remove added attributes after profiling ends
58f621d
adjust indentation in readme
bf19639
add bert example to readme
6b2cf63
upload profiler
cff1497